Pinning to normal transformers version now

allenai · Oct 3, 2024 · 4505a49 · 4505a49
1 parent 78e3a94
commit 4505a49
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/gantry-requirements.txt b/gantry-requirements.txt
@@ -31,5 +31,5 @@ peft
 wandb
 omegaconf
 s3fs
-transformers @ git+https://github.com/huggingface/transformers.git@f9b44097261270377199bb0fcbceb01fd9f7d0c0
+transformers>=4.45.1
 bitsandbytes
diff --git a/pyproject.toml b/pyproject.toml
@@ -68,7 +68,7 @@ train = [
     "omegaconf",
     "s3fs",
     "necessary",
-    "transformers @ git+https://github.com/huggingface/transformers.git@f9b44097261270377199bb0fcbceb01fd9f7d0c0"
+    "transformers>=4.45.1"
 ]
 
 [tool.setuptools.packages.find]