浏览代码

temporary fix for info

justheuristic 3 年之前
父节点
当前提交
db3ad89c2e
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      src/client/remote_sequential.py

+ 1 - 0
src/client/remote_sequential.py

@@ -102,6 +102,7 @@ class RemoteSequentialInferenceSession:
 
             # TODO begin throwaway prototype code
             remote = RemoteTransformerBlock(self.remote_sequence_info.block_infos[current_block], self.p2p)
+            _=remote.info #TODO fix
             span_uids = self.remote_sequence_info.block_uids[current_block: chosen_span.end]
             remote._info = ExpertInfo(" ".join(span_uids), chosen_span.peer_id)
             self.active_sessions.append(remote.inference_session())