Pin Transformers to 4.31.0 (#3569)

ludwig-ai · Aug 31, 2023 · cd32fb8 · cd32fb8
1 parent 63f4924
commit cd32fb8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -11,7 +11,9 @@ torchaudio
 torchtext
 torchvision
 pydantic<2.0
-transformers>=4.31.0
+# TODO(Arnav): Lift transformers package restriction once 4.32.2 is released.
+# Issue: https://github.com/ludwig-ai/ludwig/issues/3571
+transformers>=4.31.0,<4.32.0
 tokenizers>=0.13.3
 spacy>=2.3
 PyYAML>=3.12,<6.0.1,!=5.4.* #Exlude PyYAML 5.4.* due to incompatibility with awscli