Update SongGeneration_node.py

This commit is contained in:
smthemex
2025-06-23 21:20:03 +08:00
committed by GitHub
parent 181bdaaef4
commit fb4a71d61e
+1 -1
View File
@@ -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"),},)