|
@@ -22,8 +22,8 @@ __version__ = "2.3.0.dev0"
|
|
|
|
|
|
if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):
|
|
if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):
|
|
assert (
|
|
assert (
|
|
- version.parse("4.32.0") <= version.parse(transformers.__version__) < version.parse("5.0.0")
|
|
|
|
- ), "Please install a proper transformers version: pip install transformers>=4.32.0,<5.0.0"
|
|
|
|
|
|
+ version.parse("4.32.0") <= version.parse(transformers.__version__) < version.parse("4.35.0")
|
|
|
|
+ ), "Please install a proper transformers version: pip install transformers>=4.32.0,<4.35.0"
|
|
|
|
|
|
|
|
|
|
def _override_bfloat16_mode_default():
|
|
def _override_bfloat16_mode_default():
|