Add scikit-learn dependency to train langage-modeling (#21229)

2023-01-23 09:54:45 -05:00
parent 929111698c
commit 5603f78fc4
1 changed files with 1 additions and 0 deletions
--- a/examples/pytorch/language-modeling/requirements.txt
+++ b/examples/pytorch/language-modeling/requirements.txt
@@ -4,3 +4,4 @@ datasets >= 1.8.0
 sentencepiece != 0.1.92
 protobuf
 evaluate
+scikit-learn