run_generation style

2020-01-31 12:05:48 -05:00
parent ff6f1492e8
commit d18d47be67
1 changed files with 1 additions and 1 deletions
--- 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