fffiloni commited on
Commit
18bc778
·
verified ·
1 Parent(s): af98c76

upgrade torch modules

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -3
requirements.txt CHANGED
@@ -1,6 +1,9 @@
1
- torch==1.13.1
2
- torchvision==0.14.1
3
- torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
 
 
 
4
  diffusers==0.24.0
5
  transformers==4.27.0
6
  xformers==0.0.16
 
1
+ #torch==1.13.1
2
+ #torchvision==0.14.1
3
+ #torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
4
+ torch==2.1.2
5
+ torchvision==0.16.2
6
+ torchaudio==2.1.2
7
  diffusers==0.24.0
8
  transformers==4.27.0
9
  xformers==0.0.16