Spaces:
Sleeping
Sleeping
Create requirements.txt
Browse files- requirements.txt +11 -0
requirements.txt
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch>=2.0.0
|
| 2 |
+
# Install pre-built packages instead of building from source
|
| 3 |
+
--find-links https://download.pytorch.org/whl/torch_stable.html
|
| 4 |
+
--extra-index-url https://download.huggingface.co/wheels/
|
| 5 |
+
transformers>=4.31.0
|
| 6 |
+
accelerate>=0.21.0
|
| 7 |
+
gradio
|
| 8 |
+
sentencepiece
|
| 9 |
+
peft>=0.4.0
|
| 10 |
+
bitsandbytes>=0.39.0
|
| 11 |
+
trl==0.7.4
|