Skip to content

GitLab

Unverified Commit e2e77f02 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix BartModel output documentation (#7390)

parent bbb07830
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_bart.py
View file @ e2e77f02
...@@ -863,7 +863,7 @@ class BartModel(PretrainedBartModel): ...@@ -863,7 +863,7 @@ class BartModel(PretrainedBartModel):
@add_code_sample_docstrings( @add_code_sample_docstrings(
tokenizer_class=_TOKENIZER_FOR_DOC, tokenizer_class=_TOKENIZER_FOR_DOC,
checkpoint="facebook/bart-large", checkpoint="facebook/bart-large",
output_type=BaseModelOutputWithPast, output_type=Seq2SeqModelOutput,
config_class=_CONFIG_FOR_DOC, config_class=_CONFIG_FOR_DOC,
) )
def forward( def forward(
......
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