[FLax] Fix marian docs 2 (#12615)

* fix_torch_device_generate_test * remove @ * up
2021-07-09 18:28:57 +01:00
parent 165606e5b4
commit 934222e3c5
2 changed files with 2 additions and 2 deletions
--- 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