|
@@ -13,7 +13,7 @@ from hivemind.moe.server.runtime import Runtime
|
|
|
from hivemind.proto.runtime_pb2 import CompressionType
|
|
|
from hivemind.utils.logging import get_logger, use_hivemind_log_handler
|
|
|
|
|
|
-from src import BloomBlock, BloomConfig, declare_active_modules
|
|
|
+from src import BloomConfig, declare_active_modules
|
|
|
from src.bloom.from_pretrained import DTYPE_MAP, load_pretrained_block
|
|
|
from src.data_structures import CHAIN_DELIMITER, UID_DELIMITER, ServerState
|
|
|
from src.dht_utils import get_remote_module_infos
|