abyildirim commited on
Commit
4ea69df
1 Parent(s): eaab7c8

requirements.txt is made compatible with gpu

Browse files
Files changed (1) hide show
  1. 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+cpu
4
- torchvision==0.14.1+cpu
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