Vic-729 commited on
Commit
23baa3f
·
1 Parent(s): ddbf7e3
Files changed (1) hide show
  1. requirements.txt +5 -3
requirements.txt CHANGED
@@ -1,4 +1,6 @@
1
- torch==2.4.1
2
- torchvision==0.19.1
3
- torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu121
 
 
4
  transformers==4.39.0
 
1
+ einops==0.7.0
2
+ protobuf==5.26.1
3
+ torch==2.5.0
4
+ torchvision==0.20.0
5
+ torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu121
6
  transformers==4.39.0