Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -2,4 +2,5 @@ transformers==4.40.0
|
|
2 |
torch==2.2.1
|
3 |
accelerate==0.29.2
|
4 |
gradio==4.26.0
|
5 |
-
|
|
|
|
2 |
torch==2.2.1
|
3 |
accelerate==0.29.2
|
4 |
gradio==4.26.0
|
5 |
+
flash-attn>=2.5.6 # <-- Critical addition
|
6 |
+
bitsandbytes==0.43.0 # Optional for quantization
|