Update generate.py

generate.py

@@ -157,4 +157,4 @@ def generate(
         remove_invalid_values=remove_invalid_values,
         synced_gpus=synced_gpus,
         **model_kwargs,
-    )
+    )