diff --git a/setup.cfg b/setup.cfg index 417a126..7c04686 100644 --- a/setup.cfg +++ b/setup.cfg @@ -38,8 +38,8 @@ install_requires = tokenizers>=0.13.3 transformers>=4.31.0,<5.0.0 speedtest-cli==2.1.3 - pydantic>=1.10,<2.0 # 2.0 is incompatible with hivemind==1.1.8 - hivemind @ git+https://github.com/learning-at-home/hivemind + pydantic>=1.10,<2.0 # 2.0 is incompatible with hivemind yet + hivemind==1.1.9 tensor_parallel==1.0.23 humanfriendly async-timeout>=4.0.2 diff --git a/src/petals/__init__.py b/src/petals/__init__.py index c696bc6..d2f91bc 100644 --- a/src/petals/__init__.py +++ b/src/petals/__init__.py @@ -11,7 +11,7 @@ from petals.models import * from petals.utils import * from petals.utils.logging import initialize_logs as _initialize_logs -__version__ = "2.0.0.post3" +__version__ = "2.0.1" if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):