@@ -8,6 +8,7 @@ from hivemind import P2PHandlerError
from test_utils import *
import src
+from src import DistributedBloomConfig
from src.bloom.from_pretrained import load_pretrained_block
from src.client.remote_sequential import RemoteTransformerBlock
from src.data_structures import UID_DELIMITER