fix imports

This commit is contained in:
justheuristic 2022-09-01 04:04:43 +03:00
parent 3513d9ad23
commit f482a98e3a

View File

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