SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[FLax] Fix marian docs 2 (#12615)

Browse Source 
* fix_torch_device_generate_test

* remove @

* up
This commit is contained in:
Patrick von Platen
2021-07-09 18:28:57 +01:00
committed by GitHub
parent 165606e5b4
commit 934222e3c5
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/model_doc/marian.rst
View File

@@ -55,7 +55,7 @@ Tips
>>> input_ids = tokenizer(text, max_length=64, return_tensors='jax').input_ids >>> input_ids = tokenizer(text, max_length=64, return_tensors='jax').input_ids
>>> # Marian has to make use of early_stopping=True >>> # 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 Naming

2
src/transformers/models/marian/modeling_flax_marian.py
View File

@@ -1466,7 +1466,7 @@ FLAX_MARIAN_MT_DOCSTRING = """
>>> input_ids = tokenizer(text, max_length=64, return_tensors='jax').input_ids >>> input_ids = tokenizer(text, max_length=64, return_tensors='jax').input_ids
>>> # Marian has to make use of early_stopping=True >>> # 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) >>> outputs = tokenizer.batch_decode(sequences, skip_special_tokens=True)
>>> # should give `Meine Freunde sind cool, aber sie essen zu viele Kohlenhydrate.` >>> # should give `Meine Freunde sind cool, aber sie essen zu viele Kohlenhydrate.`