| from transformers import AutoConfig, AutoModel, AutoTokenizer | |
| from transformers.models.auto.modeling_auto import MODEL_MAPPING | |
| from transformers.models.auto.configuration_auto import CONFIG_MAPPING | |
| from .config import BilingualConfig | |
| from .modeling import BilingualModel, BilingualForMaskedLM, BilingualForMultipleChoice, BilingualForQuestionAnswering, BilingualForSequenceClassification, BilingualForTokenClassification | |
| AutoConfig.register("bilingual", BilingualConfig) | |
| AutoModel.register(BilingualConfig, BilingualModel) | |
| AutoModel.register(BilingualConfig, BilingualForMaskedLM) | |
| AutoModel.register(BilingualConfig, BilingualForMultipleChoice) | |
| AutoModel.register(BilingualConfig, BilingualForQuestionAnswering) | |
| AutoModel.register(BilingualConfig, BilingualForSequenceClassification) | |
| AutoModel.register(BilingualConfig, BilingualForTokenClassification) | |
| AutoTokenizer.register(BilingualConfig, AutoTokenizer) | |
| CONFIG_MAPPING.update({"bilingual": BilingualConfig}) | |
| MODEL_MAPPING.update({BilingualConfig: BilingualModel}) | |
| MODEL_MAPPING.update({BilingualConfig: BilingualForMaskedLM}) | |
| MODEL_MAPPING.update({BilingualConfig: BilingualForMultipleChoice}) | |
| MODEL_MAPPING.update({BilingualConfig: BilingualForQuestionAnswering}) | |
| MODEL_MAPPING.update({BilingualConfig: BilingualForSequenceClassification}) | |
| MODEL_MAPPING.update({BilingualConfig: BilingualForTokenClassification}) |