Hawk Lee
eff537d9dd
- Add AIIA_ASR node with FunASR (paraformer-zh + SenseVoiceSmall) - Add AIIA_Podcast_Splitter for per-speaker text splitting - Add AIIA_Podcast_Stitcher with 3-tier fuzzy alignment (exact, Levenshtein, fallback) - Register 5 previously missing modules in __init__.py - Update README with installation guide and node documentation
25 lines
714 B
Python
25 lines
714 B
Python
|
|
import sys
|
|
import os
|
|
import torch
|
|
# Add ComfyUI_AIIA to path to import voxcpm
|
|
sys.path.append("/app/ComfyUI/custom_nodes/ComfyUI_AIIA")
|
|
sys.path.append("/app/ComfyUI/custom_nodes/ComfyUI_AIIA/voxcpm_core")
|
|
|
|
from voxcpm.core import VoxCPM
|
|
|
|
try:
|
|
path = "/app/ComfyUI/models/voxcpm/VoxCPM1.5"
|
|
print(f"Loading VoxCPM from {path}")
|
|
model = VoxCPM(path, optimize=False, enable_denoiser=False)
|
|
|
|
print(f"--- DIAGNOSTICS ---")
|
|
print(f"Model VAE Sample Rate: {model.tts_model.sample_rate}")
|
|
print(f"Model VAE Config Object: {model.tts_model.audio_vae.config}")
|
|
print(f"-------------------")
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|