Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ key = os.environ.get("key")
|
|
| 12 |
from huggingface_hub import login
|
| 13 |
login(key)
|
| 14 |
from bitnet import replace_linears_in_hf
|
| 15 |
-
|
| 16 |
nf4_config = BitsAndBytesConfig(
|
| 17 |
load_in_4bit=True,
|
| 18 |
bnb_4bit_quant_type="nf4",
|
|
|
|
| 12 |
from huggingface_hub import login
|
| 13 |
login(key)
|
| 14 |
from bitnet import replace_linears_in_hf
|
| 15 |
+
os.system("pip install flash-attn --no-build-isolation")
|
| 16 |
nf4_config = BitsAndBytesConfig(
|
| 17 |
load_in_4bit=True,
|
| 18 |
bnb_4bit_quant_type="nf4",
|