diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index 7bc19ac9..e2726e50 100755 --- 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,