Spaces:
Sleeping
Sleeping
abyildirim
commited on
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
|