Hugging Face's logo

Spaces:
PSLD
/
PSLD
Sleeping

App Files Community
2
LituRout commited on
Commit
baecff8
•
1 Parent(s): d695ae8

add gpu

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu113
2
- torch
3
- torchvision
4
  git+https://github.com/huggingface/diffusers.git
5
  ftfy
6
  matplotlib
@@ -41,7 +41,7 @@ test-tube>=0.7.5
41
  streamlit>=0.73.1
42
  einops==0.3.0
43
  torch-fidelity==0.3.0
44
- transformers
45
  torchmetrics==0.6.0
46
  kornia==0.6
47
  taming-transformers-rom1504
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch==1.11.0
3
+ torchvision==0.12.0
4
  git+https://github.com/huggingface/diffusers.git
5
  ftfy
6
  matplotlib
 
41
  streamlit>=0.73.1
42
  einops==0.3.0
43
  torch-fidelity==0.3.0
44
+ transformers==4.19.2
45
  torchmetrics==0.6.0
46
  kornia==0.6
47
  taming-transformers-rom1504