Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
da3f51e
1
Parent(s):
d3fc679
Update torch requirements to 2.5.1 with cu121
Browse files- requirements.txt +5 -1
requirements.txt
CHANGED
|
@@ -1,3 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
Pillow
|
| 2 |
einops
|
| 3 |
safetensors
|
|
@@ -23,4 +27,4 @@ onnxruntime
|
|
| 23 |
accelerate>=0.25.0
|
| 24 |
gradio>=3.41.2
|
| 25 |
diffusers>=0.30.1
|
| 26 |
-
transformers>=4.46.2
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 2 |
+
torch==2.5.1
|
| 3 |
+
torchvision==0.20.1
|
| 4 |
+
torchaudio==2.5.1
|
| 5 |
Pillow
|
| 6 |
einops
|
| 7 |
safetensors
|
|
|
|
| 27 |
accelerate>=0.25.0
|
| 28 |
gradio>=3.41.2
|
| 29 |
diffusers>=0.30.1
|
| 30 |
+
transformers>=4.46.2
|