add Model2Model to __init__

2019-10-17 12:59:51 +02:00
parent c1bc709c35
commit bfb9b540d4
2 changed files with 3 additions and 16 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -87,6 +87,7 @@ if is_torch_available():
    from .modeling_distilbert import (DistilBertForMaskedLM, DistilBertModel,
                                DistilBertForSequenceClassification, DistilBertForQuestionAnswering,
                                DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
+    from .modeling_seq2seq import Model2Model

    # Optimization
    from .optimization import (AdamW, ConstantLRSchedule, WarmupConstantSchedule, WarmupCosineSchedule,