Hugging Face's logo

Spaces:
roubaofeipi
/
UltraPixel-demo
Running on Zero

App Files Community
5
roubaofeipi commited on
Commit
85ad68c
•
1 Parent(s): 1f2c3a7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  --find-links https://download.pytorch.org/whl/torch_stable.html
2
  accelerate>=0.25.0
3
- torch==2.1.2+cu118
4
- torchvision==0.16.2+cu118
5
  transformers>=4.30.0
6
  numpy>=1.23.5
7
  kornia>=0.7.0
@@ -16,4 +16,4 @@ onnxruntime>=1.16.3
16
  einops>=0.7.0
17
  onnx2torch>=1.5.13
18
  warmup-scheduler @ git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
19
- torchtools @ git+https://github.com/pabloppp/pytorch-tools
 
1
  --find-links https://download.pytorch.org/whl/torch_stable.html
2
  accelerate>=0.25.0
3
+ torch==2.0.1+cu118
4
+ torchvision==0.15.2+cu118
5
  transformers>=4.30.0
6
  numpy>=1.23.5
7
  kornia>=0.7.0
 
16
  einops>=0.7.0
17
  onnx2torch>=1.5.13
18
  warmup-scheduler @ git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
19
+ torchtools @ git+https://github.com/pabloppp/pytorch-tools