diff --git a/server/assets/workflow_yue.json b/server/assets/workflow_yue.json
index b17278d..5e3b00b 100644
--- a/server/assets/workflow_yue.json
+++ b/server/assets/workflow_yue.json
@@ -23,7 +23,7 @@
       "repetition_penalty": 1.1,
       "prompt_start_time": 0,
       "prompt_end_time": 30,
-      "max_new_tokens": 2000,
+      "max_new_tokens": 2944,
       "use_dual_tracks_prompt": false,
       "use_audio_prompt": false,
       "offload_model": true,
diff --git a/server/utils/musicWorkflow.ts b/server/utils/musicWorkflow.ts
index 6a0dc19..db2d3cc 100644
--- a/server/utils/musicWorkflow.ts
+++ b/server/utils/musicWorkflow.ts
@@ -132,7 +132,7 @@ function buildYueWorkflow(params: MusicWorkflowParams): WorkflowGraph {
       node.inputs.seed = params.seed
       node.inputs.prompt_end_time = Math.min(params.duration, 45)
       node.inputs.run_n_segment = segments
-      node.inputs.max_new_tokens = 2000
+      node.inputs.max_new_tokens = 2944
     }
   }
   applyPrefix(graph, prefix)