from transformers import PretrainedConfig class LeNetConfig(PretrainedConfig): model_type = "lenet" def __init__( self, **kwargs, ): super().__init__(**kwargs)