Upload requirements.txt with huggingface_hub
Browse files- requirements.txt +8 -0
requirements.txt
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
gradio>=4.0.0
|
| 2 |
+
torch>=2.0.0
|
| 3 |
+
transformers>=4.46.0
|
| 4 |
+
accelerate>=0.31.0
|
| 5 |
+
sentencepiece>=0.1.99
|
| 6 |
+
protobuf>=3.20.0
|
| 7 |
+
einops>=0.8.0
|
| 8 |
+
# flash-attn>=2.5.0 # Commented out as it requires CUDA and may cause issues
|