Ramji commited on
Commit
56be026
·
verified ·
1 Parent(s): b0749a3
Files changed (1) hide show
  1. requirements.txt +23 -1
requirements.txt CHANGED
@@ -4,4 +4,26 @@ transformers
4
  einops
5
  matplotlib
6
  timm
7
- accelerate
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  einops
5
  matplotlib
6
  timm
7
+ accelerate
8
+ bitsandbytes
9
+ decord
10
+ deepspeed
11
+ einops
12
+ einops-exts
13
+ huggingface_hub
14
+ imageio
15
+ numpy==1.26.4
16
+ opencv-python
17
+ orjson
18
+ peft>=0.4.0
19
+ pycocoevalcap
20
+ pyyaml
21
+ scikit-learn>=1.2.2
22
+ scipy
23
+ sentencepiece==0.1.99
24
+ shortuuid
25
+ tensorboardX
26
+ termcolor
27
+ tokenizers==0.15.1
28
+ tqdm
29
+ yacs