justheuristic 2 年之前
父節點
當前提交
f7fcb3de2b
共有 1 個文件被更改,包括 1 次插入2 次删除
  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