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
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
|
|
import sys
|
|
import os
|
|
import json
|
|
import torch
|
|
|
|
# Setup paths to mimic the node environment
|
|
sys.path.append("/app/ComfyUI/custom_nodes/ComfyUI_AIIA")
|
|
sys.path.append("/app/ComfyUI/custom_nodes/ComfyUI_AIIA/voxcpm_core")
|
|
|
|
from voxcpm.model.voxcpm import VoxCPMConfig, VoxCPMModel
|
|
from voxcpm.modules.audiovae import AudioVAE, AudioVAEConfig
|
|
|
|
path = "/app/ComfyUI/models/voxcpm/VoxCPM1.5"
|
|
config_path = os.path.join(path, "config.json")
|
|
|
|
print(f"--- DEBUGGING CONFIG LOADING ---")
|
|
try:
|
|
with open(config_path, 'r') as f:
|
|
data = json.load(f)
|
|
print(f"JSON 'audio_vae_config' sample_rate: {data.get('audio_vae_config', {}).get('sample_rate')}")
|
|
|
|
# Try loading Config Object
|
|
config = VoxCPMConfig.model_validate(data)
|
|
print(f"Loaded VoxCPMConfig.device: {config.device}")
|
|
|
|
if config.audio_vae_config:
|
|
print(f"Loaded VoxCPMConfig.audio_vae_config.sample_rate: {config.audio_vae_config.sample_rate}")
|
|
else:
|
|
print(f"Loaded VoxCPMConfig.audio_vae_config is NONE (Used Default?)")
|
|
|
|
# Try recreating the Model Load logic manually
|
|
audio_vae = AudioVAE(config.audio_vae_config)
|
|
print(f"Initialized AudioVAE.sample_rate: {audio_vae.sample_rate}")
|
|
|
|
except Exception as e:
|
|
print(f"ERROR: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|