Explorar o código

rm redundant imports

dbaranchuk %!s(int64=3) %!d(string=hai) anos
pai
achega
5c17ef3d66
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  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