Export PretrainedBartModel from __init__ (#4819)

Export PretrainedBartModel from init (#4819)
e33fdc93 · Bram Vanroy · GitHub · c58e6c12 · e33fdc93
Unverified Commit e33fdc93 authored Jun 07, 2020 by Bram Vanroy Committed by GitHub Jun 07, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/__init__.py src/transformers/__init__.py +1 -0

No files found.
--- a/src/transformers/__init__.py
+++ b/src/transformers/__init__.py
@@ -230,6 +230,7 @@ if is_torch_available():
        XLM_PRETRAINED_MODEL_ARCHIVE_LIST,
    )
    from .modeling_bart import (
+        PretrainedBartModel,
        BartForSequenceClassification,
        BartModel,
        BartForConditionalGeneration,