feat: auto-resolve synchformer checkpoint from prismaudio models dir

When synchformer_ckpt input is empty, look for synchformer_state_dict.pth
in the ComfyUI prismaudio models directory automatically.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-03-27 20:49:56 +01:00
parent b1a2ee594e
commit 93120eb6b9
+7
View File
@@ -170,6 +170,13 @@ class PrismAudioFeatureExtractor:
"--cot_text", caption_cot, "--cot_text", caption_cot,
"--output", cached_path, "--output", cached_path,
] ]
# Auto-resolve synchformer checkpoint from the prismaudio models dir
if not synchformer_ckpt:
import folder_paths
candidate = os.path.join(folder_paths.models_dir, "prismaudio", "synchformer_state_dict.pth")
if os.path.exists(candidate):
synchformer_ckpt = candidate
print(f"[PrismAudio] Auto-resolved synchformer checkpoint: {synchformer_ckpt}", flush=True)
if synchformer_ckpt: if synchformer_ckpt:
cmd.extend(["--synchformer_ckpt", synchformer_ckpt]) cmd.extend(["--synchformer_ckpt", synchformer_ckpt])