gokaygokay commited on
Commit
4f339f5
1 Parent(s): bb65761

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
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
 
1
  --find-links https://download.pytorch.org/whl/torch_stable.html
2
  accelerate>=0.25.0
3
+ torch==2.1.2
4
+ torchvision==0.16.2
5
  transformers>=4.30.0
6
  numpy>=1.23.5
7
  kornia>=0.7.0