Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Bump transformers and accelerate versions #554

Merged
merged 19 commits into from
Feb 15, 2024
Prev Previous commit
actually assert transformers version
  • Loading branch information
dvmazur committed Feb 15, 2024
commit 4a892e2b957f35e0d170e05bf7383762ea69be69
4 changes: 3 additions & 1 deletion src/petals/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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():
Expand Down
Loading