CazC commited on
Commit
1af151a
·
verified ·
1 Parent(s): ef99eb6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -6
requirements.txt CHANGED
@@ -1,9 +1,6 @@
1
-
2
- torch~=2.2
3
- torchvision==0.17.1
4
- torchaudio===2.2.1
5
- cudatoolkit==11.8
6
- pytorch-cuda==11.8
7
  omegaconf==2.3.0
8
  Pillow==10.1.0
9
  einops==0.7.0
 
1
+ torch==2.0.1
2
+ wheel
3
+ ninja
 
 
 
4
  omegaconf==2.3.0
5
  Pillow==10.1.0
6
  einops==0.7.0