diff --git a/examples/run_generation.py b/examples/run_generation.py
index 7d29aeb752..d074c9e264 100644
--- 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