Temporarily downgrade Transformers due to bug in 4.32 and up

probcomp · Sep 20, 2023 · 1faff43 · 1faff43 · gabegrand · Nov 20, 2023
1 parent 3cb8dc6
commit 1faff43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
     packages=setuptools.find_packages(),
     install_requires=['torch',
                       'numpy',
-                      'transformers',
+                      'transformers==4.30',
                       'bitsandbytes',
                       'accelerate',
                       'sentencepiece'