Spaces:
Sleeping
Sleeping
Commit
·
2868557
1
Parent(s):
bdc6b1a
Upload requirements.txt
Browse files- requirements.txt +15 -0
requirements.txt
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
mpi4py
|
| 2 |
+
numpy==1.24.2
|
| 3 |
+
datasets==2.10.1
|
| 4 |
+
peft @ git+https://github.com/huggingface/peft.git@deff03f2c251534fffd2511fc2d440e84cc54b1b
|
| 5 |
+
torch==2.0.0
|
| 6 |
+
wandb==0.14.0
|
| 7 |
+
deepspeed==0.8.3
|
| 8 |
+
trl @ git+https://github.com/lvwerra/trl.git#egg=trl-0.4.1
|
| 9 |
+
sentencepiece
|
| 10 |
+
transformers @ git+https://github.com/huggingface/transformers@c612628045822f909020f7eb6784c79700813eda
|
| 11 |
+
flask
|
| 12 |
+
flask_cors
|
| 13 |
+
icetk
|
| 14 |
+
cpm_kernels==1.0.11
|
| 15 |
+
|