from transformers import DistilBertConfig class DistilBertConfig_ANE(DistilBertConfig): model_type = "distilbert_ane" def __init__(self, **kwargs): super().__init__(**kwargs)