Spaces:
Runtime error
Runtime error
Commit
·
4ea69df
1
Parent(s):
eaab7c8
requirements.txt is made compatible with gpu
Browse files- requirements.txt +2 -3
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
-f https://download.pytorch.org/whl/torch_stable.html
|
| 2 |
git+https://github.com/openai/CLIP.git
|
| 3 |
-
torch==1.13.1+
|
| 4 |
-
torchvision==0.14.1+
|
| 5 |
pytorch-lightning==1.6.5
|
| 6 |
taming-transformers-rom1504==0.0.6
|
| 7 |
einops==0.6.0
|
|
@@ -10,5 +10,4 @@ transformers==4.27.4
|
|
| 10 |
dill==0.3.6
|
| 11 |
gradio==3.26.0
|
| 12 |
torchmetrics==0.11.4
|
| 13 |
-
gdown==4.7.1
|
| 14 |
omegaconf==2.3.0
|
|
|
|
| 1 |
-f https://download.pytorch.org/whl/torch_stable.html
|
| 2 |
git+https://github.com/openai/CLIP.git
|
| 3 |
+
torch==1.13.1+cu116
|
| 4 |
+
torchvision==0.14.1+cu116
|
| 5 |
pytorch-lightning==1.6.5
|
| 6 |
taming-transformers-rom1504==0.0.6
|
| 7 |
einops==0.6.0
|
|
|
|
| 10 |
dill==0.3.6
|
| 11 |
gradio==3.26.0
|
| 12 |
torchmetrics==0.11.4
|
|
|
|
| 13 |
omegaconf==2.3.0
|