Spaces:
Sleeping
Sleeping
Commit ·
e43cecb
1
Parent(s): 1d0a9bd
Update PyTorch dependencies for ZeroGPU compatibility
Browse files- requirements.txt +4 -4
requirements.txt
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
-
--extra-index-url https://download.pytorch.org/whl/
|
| 2 |
-
torch==2.
|
| 3 |
-
torchvision==0.
|
| 4 |
Pillow
|
| 5 |
einops
|
| 6 |
safetensors
|
|
@@ -26,4 +26,4 @@ onnxruntime
|
|
| 26 |
accelerate>=0.25.0
|
| 27 |
gradio>=3.41.2
|
| 28 |
diffusers>=0.30.1
|
| 29 |
-
transformers>=4.46.2
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu128
|
| 2 |
+
torch==2.8.0
|
| 3 |
+
torchvision==0.23.0
|
| 4 |
Pillow
|
| 5 |
einops
|
| 6 |
safetensors
|
|
|
|
| 26 |
accelerate>=0.25.0
|
| 27 |
gradio>=3.41.2
|
| 28 |
diffusers>=0.30.1
|
| 29 |
+
transformers>=4.46.2
|