justheuristic il y a 2 ans
Parent
commit
f7fcb3de2b
1 fichiers modifiés avec 1 ajouts et 2 suppressions
  1. 1 2
      tests/test_remote_sequential.py

+ 1 - 2
tests/test_remote_sequential.py

@@ -1,7 +1,6 @@
 import pytest
 import torch
-from hivemind import DHT, BatchTensorDescriptor, MSGPackSerializer, get_logger, use_hivemind_log_handler
-from hivemind.proto import runtime_pb2
+from hivemind import DHT, get_logger, use_hivemind_log_handler
 from test_utils import *
 
 from petals.bloom.from_pretrained import load_pretrained_block