run_generation style

d18d47be · Lysandre · ff6f1492 · d18d47be
Commit d18d47be authored Jan 31, 2020 by Lysandre
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/run_generation.py examples/run_generation.py +1 -1

No files found.
--- a/examples/run_generation.py
+++ b/examples/run_generation.py
@@ -221,7 +221,7 @@ def main():
        top_k=args.k,
        top_p=args.p,
        repetition_penalty=args.repetition_penalty,
-        do_sample=True
+        do_sample=True,
    )

    # Batch size == 1. to add more examples please use num_return_sequences > 1