From e5ae12b541af0b085b6c8ef5b90816a8df8efeb1 Mon Sep 17 00:00:00 2001
From: Jon Gauthier <jon@gauthiers.net>
Date: Fri, 17 Dec 2021 10:39:49 -0700
Subject: [PATCH] update transformers dependency to v4.5.0, when AutoModel was
 refactored https://github.com/huggingface/transformers/pull/11027

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 267cef9..4f520af 100644
--- 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"]),