Skip to content

GitLab

Unverified Commit e33fdc93 authored by Bram Vanroy's avatar Bram Vanroy Committed by GitHub
Browse files

Export PretrainedBartModel from __init__ (#4819)

parent c58e6c12
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/__init__.py
View file @ e33fdc93
...@@ -230,6 +230,7 @@ if is_torch_available(): ...@@ -230,6 +230,7 @@ if is_torch_available():
XLM_PRETRAINED_MODEL_ARCHIVE_LIST, XLM_PRETRAINED_MODEL_ARCHIVE_LIST,
) )
from .modeling_bart import ( from .modeling_bart import (
PretrainedBartModel,
BartForSequenceClassification, BartForSequenceClassification,
BartModel, BartModel,
BartForConditionalGeneration, BartForConditionalGeneration,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment