vivek2001123 commited on
Commit
527465d
·
1 Parent(s): 2868557

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -15
requirements.txt CHANGED
@@ -1,15 +1,9 @@
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
-
 
1
+ pip,
2
+ bitsandbytes==0.39.0,
3
+ torch==2.0.1,
4
+ transformers,
5
+ peft,
6
+ accelerate,
7
+ datasets==2.12.0,
8
+ loralib==0.1.1,
9
+ einops