riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

do_sample should be set to True in run_generation.py

Browse Source
This commit is contained in:
Lysandre 2020-01-31 11:49:32 -05:00
parent 3a21d6da6b
commit 7365f01d43
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/run_generation.py
View File

@ -221,6 +221,7 @@ def main():
top_k=args.k,
top_p=args.p,
repetition_penalty=args.repetition_penalty,
do_sample=True
)
# Batch size == 1. to add more examples please use num_return_sequences > 1