From 754202de4f7b40869626e9a4ee8990e7a3067d68 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 17 Nov 2021 19:02:33 +0100 Subject: [PATCH] [Bart] Fix docs (#14434) --- docs/source/model_doc/bart.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/model_doc/bart.rst b/docs/source/model_doc/bart.rst index 57407c0930..efd2f4bd8c 100644 --- a/docs/source/model_doc/bart.rst +++ b/docs/source/model_doc/bart.rst @@ -74,7 +74,7 @@ The :obj:`facebook/bart-base` and :obj:`facebook/bart-large` checkpoints can be .. code-block:: from transformers import BartForConditionalGeneration, BartTokenizer - model = BartForConditionalGeneration.from_pretrained("facebook/bart-large", force_bos_token_to_be_generated=True) + model = BartForConditionalGeneration.from_pretrained("facebook/bart-large", forced_bos_token_id=0) tok = BartTokenizer.from_pretrained("facebook/bart-large") example_english_phrase = "UN Chief Says There Is No in Syria" batch = tok(example_english_phrase, return_tensors='pt')