From 7365f01d437e9c7d182b82a81db8686efeb3e470 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Fri, 31 Jan 2020 11:49:32 -0500 Subject: [PATCH] do_sample should be set to True in run_generation.py --- examples/run_generation.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/run_generation.py b/examples/run_generation.py index 60838d90cc..7d29aeb752 100644 --- a/examples/run_generation.py +++ b/examples/run_generation.py @@ -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