From e92140c5677ad2c6fc3afc53248a3f5aa51a1c5b Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Thu, 2 Sep 2021 15:02:18 +0530 Subject: [PATCH] fix example (#13387) --- docs/source/model_doc/mbart.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/model_doc/mbart.rst b/docs/source/model_doc/mbart.rst index dbe1d4e435..47697844cd 100644 --- a/docs/source/model_doc/mbart.rst +++ b/docs/source/model_doc/mbart.rst @@ -49,11 +49,11 @@ inside the context manager :meth:`~transformers.MBartTokenizer.as_target_tokeniz >>> from transformers import MBartForConditionalGeneration, MBartTokenizer - >>> tokenizer = MBartTokenizer.from_pretrained("facebook/mbart-large-en-ro") + >>> tokenizer = MBartTokenizer.from_pretrained("facebook/mbart-large-en-ro", src_lang="en_XX", tgt_lang="ro_RO") >>> example_english_phrase = "UN Chief Says There Is No Military Solution in Syria" >>> expected_translation_romanian = "Şeful ONU declară că nu există o soluţie militară în Siria" - >>> inputs = tokenizer(example_english_phrase, return_tensors="pt", src_lang="en_XX", tgt_lang="ro_RO") + >>> inputs = tokenizer(example_english_phrase, return_tensors="pt") >>> with tokenizer.as_target_tokenizer(): ... labels = tokenizer(expected_translation_romanian, return_tensors="pt")