visheratin commited on
Commit
5282647
·
verified ·
1 Parent(s): cea4766

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -0
requirements.txt ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ torch>=2.0.0
2
+ transformers>=4.30.0
3
+ gradio>=3.50.0
4
+ Pillow>=9.0.0
5
+ flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
6
+ numpy>=1.20.0
7
+ tqdm>=4.64.0