|
|
@ -17,7 +17,7 @@ __version__ = "2.1.0"
|
|
|
|
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")
|
|
|
|
version.parse("4.32.0") <= version.parse(transformers.__version__) < version.parse("5.0.0")
|
|
|
|
), "Please install a proper transformers version: pip install transformers>=4.31.0,<5.0.0"
|
|
|
|
), "Please install a proper transformers version: pip install transformers>=4.32.0,<5.0.0"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _override_bfloat16_mode_default():
|
|
|
|
def _override_bfloat16_mode_default():
|
|
|
|