Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +6 -1
requirements.txt
CHANGED
@@ -5,8 +5,12 @@ torchvision==0.16.0
|
|
5 |
# Transformers for Florence-2 model
|
6 |
transformers==4.44.0
|
7 |
|
|
|
|
|
|
|
|
|
8 |
# Gradio UI framework
|
9 |
-
gradio==
|
10 |
|
11 |
# Image processing
|
12 |
pillow==10.0.1
|
@@ -21,6 +25,7 @@ spaces==0.19.4
|
|
21 |
# Optional accelerations (commented out to avoid conflicts)
|
22 |
# accelerate==0.24.1
|
23 |
# sentence-transformers==2.2.2
|
|
|
24 |
|
25 |
# Development and debugging (optional)
|
26 |
# logging
|
|
|
5 |
# Transformers for Florence-2 model
|
6 |
transformers==4.44.0
|
7 |
|
8 |
+
# Florence-2 dependencies
|
9 |
+
einops==0.8.0
|
10 |
+
timm==1.0.3
|
11 |
+
|
12 |
# Gradio UI framework
|
13 |
+
gradio==5.33.2
|
14 |
|
15 |
# Image processing
|
16 |
pillow==10.0.1
|
|
|
25 |
# Optional accelerations (commented out to avoid conflicts)
|
26 |
# accelerate==0.24.1
|
27 |
# sentence-transformers==2.2.2
|
28 |
+
# flash_attn # Optional, can work without it
|
29 |
|
30 |
# Development and debugging (optional)
|
31 |
# logging
|