From 8dcc102cd32ffce921e1890d9c9602ea2d0c020d Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sat, 22 Nov 2025 19:07:12 +0800
Subject: [PATCH] Update SongGeneration_node.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

入参顺序颠倒 导致音频输入模式无效
---
 SongGeneration_node.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/SongGeneration_node.py b/SongGeneration_node.py
index c03d0c3..6dcf394 100644
--- a/SongGeneration_node.py
+++ b/SongGeneration_node.py
@@ -77,7 +77,7 @@ class SongGeneration_Stage1:
     FUNCTION = "main"
     CATEGORY = "SongGeneration"
 
-    def main(self,vae,seperate_model, auto_prompt_audio_type,prompt_pt,model_1rvq,demucs_pt,**kwargs):
+    def main(self,vae,seperate_model,prompt_pt, auto_prompt_audio_type,model_1rvq,demucs_pt,**kwargs):
 
         audio=kwargs.get("audio", None)
         model_sep_path=folder_paths.get_full_path("SongGeneration", seperate_model) if seperate_model != "none" else None
@@ -85,7 +85,7 @@ class SongGeneration_Stage1:
         prompt_pt_path=folder_paths.get_full_path("SongGeneration", prompt_pt) if prompt_pt != "none" else None
     
         if audio is not None:
-            
+            print("Using audio as reference.")
             prompt_audio_path = os.path.join(folder_paths.get_input_directory(), f"audio_{time.strftime('%m%d%H%S')}_temp.wav")
             waveform=audio["waveform"].squeeze(0)
             buff = io.BytesIO()