From e33fdc93b4ecb571dd7a8002a74789ec8bfffc09 Mon Sep 17 00:00:00 2001 From: Bram Vanroy Date: Sun, 7 Jun 2020 17:55:10 +0200 Subject: [PATCH] Export PretrainedBartModel from __init__ (#4819) --- src/transformers/__init__.py | 1 + 1 file changed, 1 insertion(+) 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,