Browse Source

Upgrade Pydantic to >= 2.0.0 (#607)

Ink 1 năm trước cách đây
mục cha
commit
22afba627a
2 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 1 1
      setup.cfg
  2. 1 1
      src/petals/data_structures.py

+ 1 - 1
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

+ 1 - 1
src/petals/data_structures.py

@@ -2,7 +2,7 @@ import dataclasses
 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