ソースを参照

rm redundant imports

dbaranchuk 3 年 前
コミット
5c17ef3d66
1 ファイル変更2 行追加2 行削除
  1. 2 2
      src/client/sequential_autograd.py

+ 2 - 2
src/client/sequential_autograd.py

@@ -3,8 +3,8 @@ import logging
 from typing import List, Optional, Sequence, Tuple
 
 import torch
-from hivemind import deserialize_torch_tensor, serialize_torch_tensor
-from hivemind.moe.client.expert import _forward_stream, expert_backward, expert_forward
+from hivemind import serialize_torch_tensor
+from hivemind.moe.client.expert import expert_backward, expert_forward
 from hivemind.moe.client.remote_expert_worker import RemoteExpertWorker
 from hivemind.p2p import StubBase
 from hivemind.utils.nested import nested_compare, nested_flatten, nested_pack