Staticaliza commited on
Commit
03694cc
·
verified ·
1 Parent(s): 201c325

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -2
requirements.txt CHANGED
@@ -1,2 +1,12 @@
1
- transformers
2
- https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.2/flash_attn-2.6.2+cu123torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
 
 
 
 
 
 
 
 
 
 
1
+ Pillow==10.1.0
2
+ torch==2.1.2
3
+ torchvision==0.16.2
4
+ transformers==4.40.2
5
+ sentencepiece==0.1.99
6
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.2/flash_attn-2.6.2+cu123torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
7
+ opencv-python
8
+ decord
9
+ #gradio==4.22.0
10
+ gradio==4.41.0
11
+ http://thunlp.oss-cn-qingdao.aliyuncs.com/multi_modal/never_delete/modelscope_studio-0.4.0.9-py3-none-any.whl
12
+ accelerate