From fb4a71d61ec189da107e2b8f78bad238b08d9c26 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Mon, 23 Jun 2025 21:20:03 +0800
Subject: [PATCH] Update SongGeneration_node.py

---
 SongGeneration_node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/SongGeneration_node.py b/SongGeneration_node.py
index 98cbce5..be92c66 100644
--- a/SongGeneration_node.py
+++ b/SongGeneration_node.py
@@ -107,7 +107,7 @@ class SongGeneration_Stage2:
 
         descriptions=description if model.get("use_descriptions") else None
 
-        items=infer_stage2(model.get("item"),model.get("cfg"),model.get("Weigths_Path"),model.get("max_duration"),lyric,descriptions,cfg_coef, temp ,top_k ,top_p ,record_tokens ,record_window )
+        items=infer_stage2(model.get("item"),model.get("cfg"),model.get("Weigths_Path"),model.get("max_duration"),lyric,descriptions,cfg_coef, temp,top_k,top_p,record_tokens ,record_window )
         gc_clear()
         return ({"item":items,"cfg":model.get("cfg"),"max_duration":model.get("max_duration"),},)