|
|
|
@ -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():
|
|
|
|
|