Spaces:
Sleeping
Sleeping
Add torch-gpu extras dependencies for HF Spaces
Browse files- requirements.txt +4 -2
requirements.txt
CHANGED
|
@@ -1,9 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
gradio>=5.49.0
|
| 2 |
loguru>=0.7.3
|
| 3 |
memory-profiler>=0.61.0
|
| 4 |
spaces>=0.30.0
|
| 5 |
-
torch>=2.0.0
|
| 6 |
-
torchvision>=0.15.0
|
| 7 |
pandas>=2.0.0
|
| 8 |
numpy>=1.24.0
|
| 9 |
pillow>=10.0.0
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 2 |
+
torch>=2.0.0,<2.6
|
| 3 |
+
torchvision>=0.15.0
|
| 4 |
+
open-clip-torch
|
| 5 |
gradio>=5.49.0
|
| 6 |
loguru>=0.7.3
|
| 7 |
memory-profiler>=0.61.0
|
| 8 |
spaces>=0.30.0
|
|
|
|
|
|
|
| 9 |
pandas>=2.0.0
|
| 10 |
numpy>=1.24.0
|
| 11 |
pillow>=10.0.0
|