Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +11 -3
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 2 |
|
|
|
|
| 3 |
gradio==5.33.1
|
| 4 |
torch==2.1.0+cu118
|
| 5 |
torchvision==0.16.0+cu118
|
|
@@ -7,18 +8,25 @@ transformers==4.44.0
|
|
| 7 |
accelerate==0.24.1
|
| 8 |
pillow==10.0.1
|
| 9 |
requests==2.31.0
|
| 10 |
-
numpy==1.26.4
|
|
|
|
|
|
|
| 11 |
sentence-transformers==2.2.2
|
| 12 |
einops==0.8.0
|
| 13 |
timm==1.0.3
|
| 14 |
|
|
|
|
| 15 |
opencv-python==4.8.1.78
|
| 16 |
opencv-contrib-python==4.8.1.78
|
| 17 |
-
deepface==0.0.79
|
| 18 |
-
tf-keras==2.15.0
|
| 19 |
mediapipe==0.10.7
|
| 20 |
retina-face==0.0.13
|
| 21 |
gdown==4.7.1
|
| 22 |
tqdm==4.66.1
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
flash-attn==2.7.4.post1 --no-build-isolation
|
|
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 2 |
|
| 3 |
+
# Core Dependencies
|
| 4 |
gradio==5.33.1
|
| 5 |
torch==2.1.0+cu118
|
| 6 |
torchvision==0.16.0+cu118
|
|
|
|
| 8 |
accelerate==0.24.1
|
| 9 |
pillow==10.0.1
|
| 10 |
requests==2.31.0
|
| 11 |
+
numpy==1.26.4 # Avoids conflict with TensorFlow and older packages
|
| 12 |
+
|
| 13 |
+
# NLP / Embedding Models
|
| 14 |
sentence-transformers==2.2.2
|
| 15 |
einops==0.8.0
|
| 16 |
timm==1.0.3
|
| 17 |
|
| 18 |
+
# Computer Vision
|
| 19 |
opencv-python==4.8.1.78
|
| 20 |
opencv-contrib-python==4.8.1.78
|
|
|
|
|
|
|
| 21 |
mediapipe==0.10.7
|
| 22 |
retina-face==0.0.13
|
| 23 |
gdown==4.7.1
|
| 24 |
tqdm==4.66.1
|
| 25 |
|
| 26 |
+
# DeepFace & TensorFlow Support
|
| 27 |
+
deepface==0.0.79
|
| 28 |
+
tf-keras==2.15.0
|
| 29 |
+
tensorflow==2.17.0 # Pin to compatible version with deepface and CUDA 11.8
|
| 30 |
+
|
| 31 |
+
# Flash Attention (use prebuilt wheel)
|
| 32 |
flash-attn==2.7.4.post1 --no-build-isolation
|