Update to the version of transformers which supports multiple stopping criteria

256a800a · Zdenek Kasner · f777d840 · 256a800a
Commit 256a800a authored Apr 27, 2022 by Zdenek Kasner
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -29,7 +29,7 @@ setuptools.setup(
        "click>=7.1",
        "scikit-learn>=0.24.1",
        "torch>=1.7",
-        "transformers>=4.1",
+        "transformers>=4.16",
        "sqlitedict==1.6.0",
        "pytablewriter==0.58.0",
        "sacrebleu==1.5.0",