Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +1 -2
requirements.txt
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
gradio
|
| 2 |
gradio_imageslider
|
| 3 |
-
torch==2.3.0
|
| 4 |
-
torchvision==0.18.0
|
| 5 |
diffusers
|
| 6 |
transformers==4.45.0
|
| 7 |
huggingface_hub
|
|
|
|
| 1 |
gradio
|
| 2 |
gradio_imageslider
|
| 3 |
+
torch==2.3.0+cu121 # Adjust CUDA version based on your environment
|
|
|
|
| 4 |
diffusers
|
| 5 |
transformers==4.45.0
|
| 6 |
huggingface_hub
|