Tongbo commited on
Commit
a81543b
·
verified ·
1 Parent(s): b0b2441

Upload requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -1
requirements.txt CHANGED
@@ -1 +1,12 @@
1
- huggingface_hub==0.25.2
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch==2.0.1
2
+ transformers==4.38.0
3
+ sentencepiece==0.1.99
4
+ accelerate==0.23.0
5
+ peft==0.4.0
6
+ bitsandbytes==0.41.0
7
+ scikit-learn==1.2.2
8
+ einops==0.6.1
9
+ deepspeed==0.9.5
10
+ pillow
11
+ shortuuid
12
+ numpy<2.0.0