浏览代码

Add forgotten file

Aleksandr Borzunov 2 年之前
父节点
当前提交
d28cea5c84
共有 2 个文件被更改,包括 9 次插入1 次删除
  1. 1 1
      src/client/remote_model.py
  2. 8 0
      src/constants.py

+ 1 - 1
src/client/remote_model.py

@@ -15,9 +15,9 @@ from src.bloom.model import (
     BloomPreTrainedModel,
     BloomPreTrainedModel,
     LMHead,
     LMHead,
 )
 )
-from src.constants import PUBLIC_INITIAL_PEERS
 from src.client.remote_generation import RemoteGenerationMixin
 from src.client.remote_generation import RemoteGenerationMixin
 from src.client.remote_sequential import RemoteSequential
 from src.client.remote_sequential import RemoteSequential
+from src.constants import PUBLIC_INITIAL_PEERS
 from src.utils.misc import DUMMY
 from src.utils.misc import DUMMY
 
 
 use_hivemind_log_handler("in_root_logger")
 use_hivemind_log_handler("in_root_logger")

+ 8 - 0
src/constants.py

@@ -0,0 +1,8 @@
+PUBLIC_INITIAL_PEERS = [
+    "/dns/bootstrap1.petals.ml/tcp/31337/p2p/QmedTaZXmULqwspJXz44SsPZyTNKxhnnFvYRajfH7MGhCY",
+    "/dns6/bootstrap1.petals.ml/tcp/31337/p2p/QmedTaZXmULqwspJXz44SsPZyTNKxhnnFvYRajfH7MGhCY",
+    "/dns/bootstrap2.petals.ml/tcp/31338/p2p/QmQGTqmM7NKjV6ggU1ZCap8zWiyKR89RViDXiqehSiCpY5",
+    "/dns6/bootstrap2.petals.ml/tcp/31338/p2p/QmQGTqmM7NKjV6ggU1ZCap8zWiyKR89RViDXiqehSiCpY5",
+    "/dns/bootstrap3.petals.ml/tcp/31339/p2p/QmX82nfE57CSkNgyEC7pPMPBzjcFLLJXdHhvp1AXKVPvJD",
+    "/dns6/bootstrap3.petals.ml/tcp/31339/p2p/QmX82nfE57CSkNgyEC7pPMPBzjcFLLJXdHhvp1AXKVPvJD",
+]