versions check

pull/554/head
Denis Mazur 4 months ago
parent 3dab138f57
commit 103b437770

@ -17,14 +17,11 @@ from petals.models import *
from petals.utils import *
from petals.utils.logging import initialize_logs as _initialize_logs
__version__ = "2.3.0.dev1"
__version__ = "2.3.0.dev2"
if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):
# assert (
# 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"
...
version.parse(transformers.__version__) == version.parse("4.37.1")
def _override_bfloat16_mode_default():

Loading…
Cancel
Save