Junfeng5 commited on
Commit
fc8fc89
·
verified ·
1 Parent(s): caf313c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,6 +1,8 @@
1
  torch==2.4.0
2
  torchvision
3
  transformers==4.39.2
 
 
4
  timm==1.0.8
5
  deepspeed==0.12.6
6
  numpy==1.26.4
 
1
  torch==2.4.0
2
  torchvision
3
  transformers==4.39.2
4
+ sentencepiece==0.1.99
5
+ scikit-learn==1.2.2
6
  timm==1.0.8
7
  deepspeed==0.12.6
8
  numpy==1.26.4