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