Update CodeQwen1.5-base-Chat.py

QwenLM · Apr 16, 2024 · 3cb66ef · 3cb66ef
1 parent 70b05f0
commit 3cb66ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/CodeQwen1.5-base-Chat.py b/examples/CodeQwen1.5-base-Chat.py
@@ -26,7 +26,7 @@
 # Use `max_new_tokens` to control the maximum output length.
 generated_ids = model.generate(
     model_inputs.input_ids,
-    max_new_tokens=2048
+    max_new_tokens=2048 # can increase the output length
 )
 generated_ids = [
     output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)