diff --git a/setup.cfg b/setup.cfg index 857120df..f86ebc4f 100644 --- a/setup.cfg +++ b/setup.cfg @@ -39,7 +39,7 @@ install_requires = tokenizers>=0.13.3 transformers==4.43.1 # if you change this, please also change version assert in petals/__init__.py speedtest-cli==2.1.3 - hivemind @ git+https://github.com/learning-at-home/hivemind.git@f98acba7aa3cf93717c40ca95d0b4e5b5742414c + hivemind @ git+https://github.com/learning-at-home/hivemind.git@213bff98a62accb91f254e2afdccbf1d69ebdea9 tensor_parallel==1.0.23 humanfriendly async-timeout>=4.0.2 diff --git a/src/petals/data_structures.py b/src/petals/data_structures.py index 9cbbf76e..1c500f89 100644 --- a/src/petals/data_structures.py +++ b/src/petals/data_structures.py @@ -2,7 +2,7 @@ from enum import Enum from typing import Any, Dict, Optional, Sequence, Tuple -import pydantic +import pydantic.v1 as pydantic from hivemind import PeerID from hivemind.moe.expert_uid import ExpertUID