From d59464db6bb0cd2de4c95dd587cc5a66f3209cf6 Mon Sep 17 00:00:00 2001 From: Oren Amsalem Date: Mon, 8 Mar 2021 12:15:06 +0200 Subject: [PATCH] fix BART Summarization example in doc (#10582) --- src/transformers/models/bart/modeling_bart.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/bart/modeling_bart.py b/src/transformers/models/bart/modeling_bart.py index 1d15fb735e..da94fc894b 100755 --- a/src/transformers/models/bart/modeling_bart.py +++ b/src/transformers/models/bart/modeling_bart.py @@ -529,8 +529,8 @@ BART_GENERATION_EXAMPLE = r""" >>> from transformers import BartTokenizer, BartForConditionalGeneration, BartConfig - >>> model = BartForConditionalGeneration.from_pretrained('facebook/bart-large') - >>> tokenizer = BartTokenizer.from_pretrained('facebook/bart-large') + >>> model = BartForConditionalGeneration.from_pretrained('facebook/bart-large-cnn') + >>> tokenizer = BartTokenizer.from_pretrained('facebook/bart-large-cnn') >>> ARTICLE_TO_SUMMARIZE = "My friends are cool but they eat too many carbs." >>> inputs = tokenizer([ARTICLE_TO_SUMMARIZE], max_length=1024, return_tensors='pt')