瀏覽代碼

import policy

justheuristic 2 年之前
父節點
當前提交
952514eaab
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      src/petals/client/sequence_manager.py

+ 1 - 0
src/petals/client/sequence_manager.py

@@ -10,6 +10,7 @@ from hivemind.proto import runtime_pb2
 from hivemind.utils.logging import get_logger, use_hivemind_log_handler
 
 import petals.dht_utils
+from petals.client import NoSpendingPolicy
 from petals.data_structures import ModuleUID, RemoteModuleInfo, RemoteSpanInfo, ServerState
 from petals.server.handler import TransformerConnectionHandler