tsujuifu commited on
Commit
2f6d8d9
β€’
1 Parent(s): 3748098

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- -i https://download.pytorch.org/whl/cu118
2
- torch==2.0
3
- torchvision==0.15
4
- torchaudio==2.0
 
1
+ -i https://download.pytorch.org/whl/cu121
2
+ torch==2.2.2
3
+ torchvision==0.17.2
4
+ torchaudio==2.2.2