Clean special token init in modeling_....py (#3264)

* make style * fix conflicts
2020-03-20 21:41:04 +01:00
parent 8becb73293
commit 95e00d0808
22 changed files with 117 additions and 115 deletions
--- a/src/transformers/init.py
+++ b/src/transformers/init.py
@@ -223,6 +223,7 @@ if is_torch_available():
        BartForSequenceClassification,
        BartModel,
        BartForConditionalGeneration,
+        BART_PRETRAINED_MODEL_ARCHIVE_MAP,
    )
    from .modeling_roberta import (
        RobertaForMaskedLM,