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:
@@ -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])
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user