Upload requirements.txt
Browse files- requirements.txt +4 -2
requirements.txt
CHANGED
|
@@ -1,5 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
gradio
|
| 2 |
-
torch
|
| 3 |
diffusers
|
| 4 |
transformers
|
| 5 |
accelerate
|
|
@@ -14,4 +16,4 @@ modelscope
|
|
| 14 |
spaces
|
| 15 |
simplejson
|
| 16 |
sortedcontainers
|
| 17 |
-
timm
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu128
|
| 2 |
+
torch==2.9.0+cu128
|
| 3 |
+
torchvision==0.24.0+cu128
|
| 4 |
gradio
|
|
|
|
| 5 |
diffusers
|
| 6 |
transformers
|
| 7 |
accelerate
|
|
|
|
| 16 |
spaces
|
| 17 |
simplejson
|
| 18 |
sortedcontainers
|
| 19 |
+
timm
|