YiftachEde commited on
Commit
9e32a0a
·
1 Parent(s): 0982ce8
Files changed (1) hide show
  1. requirements.txt +2 -3
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch>=2.1.0
2
- torchvision>=0.16.0
3
  gradio>=4.13.0,<5.0.0
4
  diffusers
5
  transformers
@@ -20,4 +20,3 @@ git+https://github.com/NVlabs/nvdiffrast.git
20
  rembg
21
  onnxruntime
22
  kiui
23
- --extra-index-url https://download.pytorch.org/whl/cu118
 
1
+ torch --index-url https://download.pytorch.org/whl/cu118
2
+ torchvision --index-url https://download.pytorch.org/whl/cu118
3
  gradio>=4.13.0,<5.0.0
4
  diffusers
5
  transformers
 
20
  rembg
21
  onnxruntime
22
  kiui