From 4a892e2b957f35e0d170e05bf7383762ea69be69 Mon Sep 17 00:00:00 2001
From: Denis Mazur <denismazur8@gmail.com>
Date: Thu, 15 Feb 2024 11:21:03 +0300
Subject: [PATCH] actually assert transformers version

---
 src/petals/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/petals/__init__.py b/src/petals/__init__.py
index 329783e2b..fd3893644 100644
--- 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():