File size: 279 Bytes
03ea185 |
1 2 3 4 5 6 7 |
from transformers import AutoConfig, AutoModel
from .configuration_mic21 import MIC21SummarizerConfig
from .modeling_mic21 import MIC21SummarizerModel
AutoConfig.register("mic21_summarizer", MIC21SummarizerConfig)
AutoModel.register(MIC21SummarizerConfig, MIC21SummarizerModel)
|