cerebras.modelzoo.config.utils.create_config_class#
- cerebras.modelzoo.config.utils.create_config_class(cls, custom_type_mapping=None, **kwargs)[source]#
Dynamically construct a BaseConfig class from the signature of a class/function.
This function will inspect the signature of the provided cls and use the annotations to create a Pydantic model.
- Parameters
cls (type) – The class or function to create a config class for.
custom_type_mapping (Optional[Dict[type, type]]) – A dictionary mapping annotations to Pydantic types.
**kwargs – Additional keyword arguments to pass to create_model.