Sergidev commited on
Commit
e6b6132
·
verified ·
1 Parent(s): 29ca3eb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +17 -11
requirements.txt CHANGED
@@ -1,12 +1,18 @@
1
- torch>=2.0.0
2
- transformers>=4.33.0
3
- accelerate>=0.21.0
4
- peft>=0.4.0
5
- datasets>=2.12.0
6
- sentencepiece>=0.1.99
7
- protobuf>=4.24.0
8
- tiktoken>=0.4.0
9
- scipy
10
- vllm>=0.2.0
11
  cn2an>=0.5.22
12
- langdetect>=1.0.9
 
 
 
 
 
 
 
1
+ transformers==4.43.2
2
+ vllm==0.5.5
3
+ vllm-flash-attn
4
+ flash_attn
5
+ deepspeed==0.14.4
6
+ llamafactory
7
+ openai
8
+ datasets
9
+ tqdm
10
+ gradio
11
  cn2an>=0.5.22
12
+ langdetect>=1.0.9
13
+ accelerate
14
+ peft
15
+ sentencepiece
16
+ protobuf
17
+ tiktoken
18
+ scipy