fix imports

remove-remote-block
justheuristic 2 years ago
parent 3513d9ad23
commit f482a98e3a

@ -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

Loading…
Cancel
Save