Hugging Face's logo

Spaces:
shi-labs
/
FcF-Inpainting
Runtime error

App Files Community
4
praeclarumjj3 commited on
Commit
3291051
•
1 Parent(s): b170e0d

Update setup.sh

Browse files
Files changed (1) hide show
  1. setup.sh +2 -0
setup.sh CHANGED
@@ -1,4 +1,6 @@
1
  #!/bin/sh
2
  pip3 uninstall torch
 
 
3
  pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
4
  python3 -c "import torch; print(torch.__version__)"
 
1
  #!/bin/sh
2
  pip3 uninstall torch
3
+ pip3 uninstall torch
4
+ pip3 install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
5
  pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
6
  python3 -c "import torch; print(torch.__version__)"