Update requirements.txt
Browse files- requirements.txt +4 -3
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
-
torch>=2.
|
| 2 |
transformers>=4.35.0
|
| 3 |
-
accelerate>=0.
|
| 4 |
gradio>=4.0.0
|
| 5 |
-
safetensors>=0.4.0
|
|
|
|
|
|
| 1 |
+
torch>=2.1.0
|
| 2 |
transformers>=4.35.0
|
| 3 |
+
accelerate>=0.24.0
|
| 4 |
gradio>=4.0.0
|
| 5 |
+
safetensors>=0.4.0
|
| 6 |
+
flash-attn --no-build-isolation # Fast attention ke liye
|