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