|
@@ -15,9 +15,9 @@ from src.bloom.model import (
|
|
BloomPreTrainedModel,
|
|
BloomPreTrainedModel,
|
|
LMHead,
|
|
LMHead,
|
|
)
|
|
)
|
|
-from src.constants import PUBLIC_INITIAL_PEERS
|
|
|
|
from src.client.remote_generation import RemoteGenerationMixin
|
|
from src.client.remote_generation import RemoteGenerationMixin
|
|
from src.client.remote_sequential import RemoteSequential
|
|
from src.client.remote_sequential import RemoteSequential
|
|
|
|
+from src.constants import PUBLIC_INITIAL_PEERS
|
|
from src.utils.misc import DUMMY
|
|
from src.utils.misc import DUMMY
|
|
|
|
|
|
use_hivemind_log_handler("in_root_logger")
|
|
use_hivemind_log_handler("in_root_logger")
|