black-isort
parent
01b9bced78
commit
d03b38b9eb
@ -1,3 +1,5 @@
|
||||
from .bloom import *
|
||||
from .client import RemoteTransformerBlock, RemoteTransformerBlockInferenceSession
|
||||
from .dht_utils import declare_active_modules, get_remote_module
|
||||
|
||||
__version__ = "0.1"
|
||||
|
@ -1,8 +1,8 @@
|
||||
from typing import NamedTuple, Collection
|
||||
from typing import Collection, NamedTuple
|
||||
|
||||
from hivemind import PeerID
|
||||
|
||||
|
||||
ModuleUID = str
|
||||
UID_DELIMITER = '.'
|
||||
UID_DELIMITER = "." # delimits parts of one module uid, e.g. "bloom.transformer.h.4.self_attention"
|
||||
CHAIN_DELIMITER = " " # delimits multiple uids in a sequence, e.g. "bloom.layer3 bloom.layer4"
|
||||
RemoteModuleInfo = NamedTuple("RemoteModuleInfo", [("uid", ModuleUID), ("peer_ids", Collection[PeerID])])
|
||||
|
Loading…
Reference in New Issue