BestWishYsh commited on
Commit
7706599
·
verified ·
1 Parent(s): 17c1ed4

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- torch
2
- torchaudio
3
- torchvision
4
- xformers
5
  onnx==1.17.0
6
  onnxruntime-gpu==1.19.2
7
  deepspeed==0.15.2
 
1
+ torch==2.5.1
2
+ torchaudio==2.5.1
3
+ torchvision==0.20.1
4
+ xformers==0.0.28.post3
5
  onnx==1.17.0
6
  onnxruntime-gpu==1.19.2
7
  deepspeed==0.15.2