Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +6 -7
requirements.txt
CHANGED
@@ -1,9 +1,8 @@
|
|
1 |
-
#
|
2 |
-
|
3 |
-
|
4 |
-
torchvision
|
5 |
|
6 |
-
#
|
7 |
numpy==1.24.4
|
8 |
Pillow==10.3.0
|
9 |
Requests==2.31.0
|
@@ -13,6 +12,6 @@ qwen-vl-utils
|
|
13 |
gradio==4.44.1
|
14 |
bitsandbytes>=0.41.0
|
15 |
|
16 |
-
#
|
17 |
-
--
|
18 |
flash-attn==2.5.5
|
|
|
1 |
+
# Core dependencies
|
2 |
+
torch>=2.0.0
|
3 |
+
torchvision>=0.15.0
|
|
|
4 |
|
5 |
+
# Other dependencies
|
6 |
numpy==1.24.4
|
7 |
Pillow==10.3.0
|
8 |
Requests==2.31.0
|
|
|
12 |
gradio==4.44.1
|
13 |
bitsandbytes>=0.41.0
|
14 |
|
15 |
+
# Pre-built wheel for flash-attn
|
16 |
+
--find-links https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.5/
|
17 |
flash-attn==2.5.5
|