Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
@@ -1,9 +1,10 @@
|
|
1 |
-
torch==2.
|
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
|