Docs: fix example failing doctest in generation_strategies.md (#25874)

This commit is contained in:
Joao Gante 2023-08-30 16:23:44 +01:00 committed by GitHub
parent 72298178bc
commit 459bc6738c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -332,7 +332,8 @@ The diversily penalty ensures the outputs are distinct across groups, and beam s
>>> outputs = model.generate(**inputs, num_beams=5, num_beam_groups=5, max_new_tokens=30, diversity_penalty=1.0) >>> outputs = model.generate(**inputs, num_beams=5, num_beam_groups=5, max_new_tokens=30, diversity_penalty=1.0)
>>> tokenizer.decode(outputs[0], skip_special_tokens=True) >>> tokenizer.decode(outputs[0], skip_special_tokens=True)
'The aim of this project is to create a new type of living system, one that is more sustainable and efficient than the current one.' 'The Design Principles are a set of universal design principles that can be applied to any location, climate and
culture, and they allow us to design the'
``` ```
This guide illustrates the main parameters that enable various decoding strategies. More advanced parameters exist for the This guide illustrates the main parameters that enable various decoding strategies. More advanced parameters exist for the