From 3cb66ef778a207b86c2c67d09e240c5464b860b0 Mon Sep 17 00:00:00 2001
From: CSJianYang <506733299@qq.com>
Date: Wed, 17 Apr 2024 00:08:54 +0800
Subject: [PATCH] Update CodeQwen1.5-base-Chat.py

---
 examples/CodeQwen1.5-base-Chat.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/CodeQwen1.5-base-Chat.py b/examples/CodeQwen1.5-base-Chat.py
index 44ffe3c..5553a1a 100644
--- 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)