o Ti@s`ddlmZmZddlmZmZmZddlmZddl m Z GdddeZ Gdd d eZ d S) )ABCabstractstaticmethod)AnyDictType)DeepSpeedConfigModel) DSModuleBasec@s6eZdZUdZeed<eed<iZeee fed<dS) ConfigBundlezi A config bundle is a collection of configs that are used to instantiate a model implementation. nameconfigimplementation_configN) __name__ __module__ __qualname____doc__str__annotations__rr rrrrb/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/inference/v2/modules/module_registry.pyr s r c@sReZdZdZededefddZede efddZ ededd fd d Z d S) DSModuleRegistryBasez] Class holding logic for tracking the DSModule implementations of a given interface. config_bundlereturncCsh|j|jvrtd|jd|j|j|j}||js)td|jd||j|j|j|jS)z> Given a DSModule key, attempt to instantiate zUnknown DSModule: z, cls.registry=zConfig z is not supported by )r registryKeyErrorsupports_configr ValueErrorr )clsrtarget_implementationrrrinstantiate_configs   z'DSModuleRegistryBase.instantiate_configcCstd)zA Return the class associated with this registry. z3Must associated a DSModule class with its registry.)NotImplementedErrorrrrrassociated_class)sz%DSModuleRegistryBase.associated_class child_classNcCsBt||std|d|d|||j|<|S)z7 Register a module with this registry. z Can only register subclasses of z, z does not inherit from ) issubclassr! TypeErrorrr )rr"rrrregister_module0s z$DSModuleRegistryBase.register_module) rrrr classmethodr r rrrr!r%rrrrrs rN) abcrrtypingrrrdeepspeed.runtime.config_utilsr ds_moduler r rrrrrs