| 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) | |