Browse Source

undo refactor

justheuristic 3 years ago
parent
commit
475c63db98
1 changed files with 1 additions and 2 deletions
  1. 1 2
      src/client/remote_sequential.py

+ 1 - 2
src/client/remote_sequential.py

@@ -12,8 +12,7 @@ from hivemind.moe.expert_uid import ExpertInfo
 from torch import nn
 from torch import nn
 
 
 import src
 import src
-from src.client.remote_block import RemoteTransformerBlock
-from src import RemoteTransformerBlockInferenceSession
+from src.client.remote_block import RemoteTransformerBlock, RemoteTransformerBlockInferenceSession
 from src.client.sequence_manager import RemoteSequenceManager
 from src.client.sequence_manager import RemoteSequenceManager
 from src.data_structures import UID_DELIMITER, RemoteSpanInfo
 from src.data_structures import UID_DELIMITER, RemoteSpanInfo
 from src.dht_utils import _create_remote_modules_from_infos
 from src.dht_utils import _create_remote_modules_from_infos