diff --git a/src/transformers/modeling_bart.py b/src/transformers/modeling_bart.py index d4f93b919c6..f1dca793db2 100644 --- a/src/transformers/modeling_bart.py +++ b/src/transformers/modeling_bart.py @@ -863,7 +863,7 @@ class BartModel(PretrainedBartModel): @add_code_sample_docstrings( tokenizer_class=_TOKENIZER_FOR_DOC, checkpoint="facebook/bart-large", - output_type=BaseModelOutputWithPast, + output_type=Seq2SeqModelOutput, config_class=_CONFIG_FOR_DOC, ) def forward(