Junfeng5 commited on
Commit
ed7a89e
·
verified ·
1 Parent(s): 8f46170

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,9 +1,10 @@
1
- torch==2.4.0
2
  transformers==4.39.2
3
  spaces
4
  gradio==4.44.1
5
  gradio_client==1.3.0
6
  pillow
7
  accelerate
8
- bitsandbytes
9
  tqdm
 
 
1
+ torch==2.1.0
2
  transformers==4.39.2
3
  spaces
4
  gradio==4.44.1
5
  gradio_client==1.3.0
6
  pillow
7
  accelerate
8
+ bitsandbytes==4.41.0
9
  tqdm
10
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu122torch2.1cxx11abiFALSE-cp39-cp39-linux_x86_64.whl