Jose Benitez commited on
Commit
5ace994
·
1 Parent(s): 1b73739

update reqs

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -20
requirements.txt CHANGED
@@ -1,26 +1,8 @@
1
  torch==2.0.1
 
2
  deepspeed
3
  sentencepiece
4
  accelerate
5
- evaluate
6
- datasets
7
- sacrebleu
8
- websockets
9
- omegaconf
10
- scikit-learn
11
- jiwer
12
- tenacity
13
- pynvml
14
- ninja
15
- ray
16
- psutil
17
- xformers >= 0.0.19
18
- fastapi
19
- geomloss
20
- einops
21
- langdetect
22
- vllm==0.1.4
23
  transformers
24
  transformers_stream_generator
25
- plotly
26
- flash_attn
 
1
  torch==2.0.1
2
+ flash_attn
3
  deepspeed
4
  sentencepiece
5
  accelerate
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  transformers
7
  transformers_stream_generator
8
+ plotly