__init__.py

# ---------------------------------------------------------------------------
# Register the model with the Transformers library
# ---------------------------------------------------------------------------
from transformers import AutoConfig, AutoModel, AutoModelForSequenceClassification
from .configuration_bert_ordinal import BertOrdinalConfig
from .modeling_bert_ordinal import BertOrdinal

AutoConfig.register("bert_ordinal", BertOrdinalConfig)
AutoModel.register(BertOrdinalConfig, BertOrdinal)
AutoModelForSequenceClassification.register(BertOrdinalConfig, BertOrdinal)