From 934222e3c572e37a8f15660876152c7df5cd7429 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 9 Jul 2021 18:28:57 +0100 Subject: [PATCH] [FLax] Fix marian docs 2 (#12615) * fix_torch_device_generate_test * remove @ * up --- docs/source/model_doc/marian.rst | 2 +- src/transformers/models/marian/modeling_flax_marian.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/model_doc/marian.rst b/docs/source/model_doc/marian.rst index ce8c2d0c42..41c5d67198 100644 --- a/docs/source/model_doc/marian.rst +++ b/docs/source/model_doc/marian.rst @@ -55,7 +55,7 @@ Tips >>> input_ids = tokenizer(text, max_length=64, return_tensors='jax').input_ids >>> # Marian has to make use of early_stopping=True - >>> sequences = model.generate(inputs, early_stopping=True, max_length=64, num_beams=2).sequences + >>> sequences = model.generate(input_ids, early_stopping=True, max_length=64, num_beams=2).sequences Naming diff --git a/src/transformers/models/marian/modeling_flax_marian.py b/src/transformers/models/marian/modeling_flax_marian.py index 76bd7388bd..18476ff93d 100644 --- a/src/transformers/models/marian/modeling_flax_marian.py +++ b/src/transformers/models/marian/modeling_flax_marian.py @@ -1466,7 +1466,7 @@ FLAX_MARIAN_MT_DOCSTRING = """ >>> input_ids = tokenizer(text, max_length=64, return_tensors='jax').input_ids >>> # Marian has to make use of early_stopping=True - >>> sequences = model.generate(inputs, early_stopping=True, max_length=64, num_beams=2).sequences + >>> sequences = model.generate(input_ids, early_stopping=True, max_length=64, num_beams=2).sequences >>> outputs = tokenizer.batch_decode(sequences, skip_special_tokens=True) >>> # should give `Meine Freunde sind cool, aber sie essen zu viele Kohlenhydrate.`