Apiarist Dev commited on
Commit ·
3d9702c
1
Parent(s): 2c5f41c
fix: bump torch to 2.8.0 for ZeroGPU compatibility, restore CUDA wheel
Browse files- requirements.txt +2 -4
requirements.txt
CHANGED
|
@@ -1,9 +1,7 @@
|
|
| 1 |
-
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 2 |
-
|
| 3 |
gradio>=5.0.0
|
| 4 |
Pillow>=10.0.0
|
| 5 |
transformers>=4.40.0,<5.0.0
|
| 6 |
-
torch==2.
|
| 7 |
-
torchvision==0.
|
| 8 |
einops
|
| 9 |
accelerate
|
|
|
|
|
|
|
|
|
|
| 1 |
gradio>=5.0.0
|
| 2 |
Pillow>=10.0.0
|
| 3 |
transformers>=4.40.0,<5.0.0
|
| 4 |
+
torch==2.8.0
|
| 5 |
+
torchvision==0.23.0
|
| 6 |
einops
|
| 7 |
accelerate
|