Spaces:
Paused
Paused
Wenye He
commited on
Update requirements.txt
Browse files- requirements.txt +6 -4
requirements.txt
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
-
torch
|
| 2 |
-
transformers
|
| 3 |
-
|
| 4 |
-
|
|
|
|
|
|
|
|
|
| 1 |
+
torch>=2.1.2
|
| 2 |
+
transformers>=4.40.0
|
| 3 |
+
gradio>=4.24.0
|
| 4 |
+
accelerate>=0.29.0
|
| 5 |
+
bitsandbytes>=0.43.0
|
| 6 |
+
flash-attn>=2.5.6 # Needed for Phi-3 optimization
|