actually assert transformers version

bigscience-workshop · Feb 15, 2024 · 4a892e2 · 4a892e2
1 parent 103b437
commit 4a892e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/petals/__init__.py b/src/petals/__init__.py
@@ -21,7 +21,9 @@
 
 
 if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):
-    version.parse(transformers.__version__) == version.parse("4.37.1")
+    assert (
+        version.parse("4.37.1") <= version.parse(transformers.__version__) < version.parse("4.38.0")
+    ), "Please install a proper transformers version: pip install transformers>=4.37.1,<4.38.0"
 
 
 def _override_bfloat16_mode_default():