from transformers import PretrainedConfig


class CVRRMergedConfig(PretrainedConfig):
    model_type = 'cvrr_merged'

    def __init__(self, release=None, **kwargs):
        super().__init__(**kwargs)
        self.release = {} if release is None else release
        self.architectures = ['CVRRMergedModel']
        self.auto_map = {
            'AutoConfig': 'configuration_cvrr_merged.CVRRMergedConfig',
            'AutoModel': 'modeling_cvrr_merged.CVRRMergedModel',
            'AutoModelForImageTextToText': 'modeling_cvrr_merged.CVRRMergedModel',
        }


CVRRMergedConfig.register_for_auto_class()