riabayonaor commited on
Commit
49970ad
·
verified ·
1 Parent(s): 2ab855f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +19 -3
requirements.txt CHANGED
@@ -1,3 +1,19 @@
1
- transformers
2
- gradio
3
- torch
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch==2.0.1
2
+ transformers>=4.38.2
3
+ timm>=0.9.16
4
+ accelerate
5
+ sentencepiece
6
+ attrdict
7
+ einops
8
+
9
+ # for gradio demo
10
+ gradio==3.48.0
11
+ gradio-client==0.6.1
12
+ mdtex2html==1.3.0
13
+ pypinyin==0.50.0
14
+ tiktoken==0.5.2
15
+ tqdm==4.64.0
16
+ colorama==0.4.5
17
+ Pygments==2.12.0
18
+ markdown==3.4.1
19
+ SentencePiece==0.1.96