fffiloni commited on
Commit
7834e6c
·
verified ·
1 Parent(s): d59f323

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,3 +1,5 @@
 
 
1
  transformers==4.42.3
2
  xtuner[deepspeed]==0.1.23
3
  timm==1.0.9
 
1
+ torch==2.3.1
2
+ torchvision==0.18.1
3
  transformers==4.42.3
4
  xtuner[deepspeed]==0.1.23
5
  timm==1.0.9