Update requirements

ModelTC · Aug 23, 2024 · c7fbb57 · c7fbb57
1 parent 06b1658
commit c7fbb57
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/requirements/runtime.txt b/requirements/runtime.txt
@@ -1,10 +1,24 @@
-torch
+torch>=2.1.0
 loguru
-transformers==4.41.2
-datasets
+transformers>=4.41.2
 huggingface-hub
 sentencepiece
 protobuf
-accelerate
+accelerate>=0.26.0
 zstandard
 easydict
+evaluate>=0.4.0
+datasets>=2.16.0
+jsonlines
+numexpr
+peft>=0.2.0
+pybind11>=2.6.2
+pytablewriter
+rouge-score>=0.0.4
+sacrebleu>=1.5.0
+scikit-learn>=0.24.1
+sqlitedict
+tqdm-multiprocess
+dill
+word2number
+more_itertools