temporary fix for info

2 years ago · db3ad89c2e
parent 6113a7e955
commit db3ad89c2e
1 changed files with 1 additions and 0 deletions
--- a/src/client/remote_sequential.py
+++ b/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())