Fix BartModel output documentation (#7390)

e2e77f02 · Sylvain Gugger · GitHub · bbb07830 · e2e77f02
Unverified Commit e2e77f02 authored Sep 25, 2020 by Sylvain Gugger Committed by GitHub Sep 25, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/modeling_bart.py src/transformers/modeling_bart.py +1 -1

No files found.
--- 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(