transformers.py

from .modeling_ensemble_deberta import EnsembleDeberta, EnsembleDebertaConfig
from transformers import AutoConfig, AutoModel

# Register the custom config and model with the auto classes.
AutoConfig.register("ensemble-deberta", EnsembleDebertaConfig)
AutoModel.register(EnsembleDebertaConfig, EnsembleDeberta)