Rename BartForMaskedLM -> BartForConditionalGeneration (#3114)

* improved documentation
2020-03-05 17:41:18 -05:00
parent fa2aa699da
commit 857e0a0d3b
7 changed files with 75 additions and 71 deletions
--- a/src/transformers/init.py
+++ b/src/transformers/init.py
@@ -206,7 +206,11 @@ if is_torch_available():
        XLMForQuestionAnsweringSimple,
        XLM_PRETRAINED_MODEL_ARCHIVE_MAP,
    )
-    from .modeling_bart import BartForSequenceClassification, BartModel, BartForMaskedLM
+    from .modeling_bart import (
+        BartForSequenceClassification,
+        BartModel,
+        BartForConditionalGeneration,
+    )
    from .modeling_roberta import (
        RobertaForMaskedLM,
        RobertaModel,