Spaces:

bigscience
/

petals-api

Runtime error

petals-api / src /data_structures.py

younesbelkada

first files(

5bdad4f almost 2 years ago

No virus

391 Bytes

	from typing import Collection, NamedTuple

	from hivemind import PeerID

	ModuleUID = str
	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])])