KaiChen1998 commited on
Commit
e8e2c8b
·
verified ·
1 Parent(s): 3c042eb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  omegaconf
2
- # torch
3
- # torchvision
4
  transformers==4.44.2
5
  sentencepiece==0.1.99
6
  accelerate==0.33.0
 
1
  omegaconf
2
+ torch==2.4.0
3
+ torchvision==0.19.0
4
  transformers==4.44.2
5
  sentencepiece==0.1.99
6
  accelerate==0.33.0