Amossofer commited on
Commit
feba089
·
1 Parent(s): f45c0a2
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -1,3 +1,6 @@
1
- gradio>=3.50.0
2
  transformers>=4.40.0
3
- torch>=2.2.0
 
 
 
 
1
+ gradio>=4.18.0
2
  transformers>=4.40.0
3
+ torch>=2.1.0
4
+ accelerate>=0.27.0
5
+ sentencepiece # required for some tokenizers
6
+ safetensors # faster & safer model loading