update transformers dependency to v4.5.0, when AutoModel was refactored

huggingface/transformers#11027
cpllab · Dec 17, 2021 · e5ae12b · e5ae12b
1 parent 3f1c53d
commit e5ae12b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
     extras_require={
         "docker": ["docker~=5.0.3"],
         "singularity": ["spython~=0.1.17"],
-        "huggingface": ["transformers~=4.12", "torch"],
+        "huggingface": ["transformers >= 4.5, < 5.0", "torch"],
     },
     name="lm-zoo",
     packages=find_packages(exclude=["test"]),