wildminder/ComfyUI-VibeVoice
1
0
Fork 0
mirror of https://github.com/wildminder/ComfyUI-VibeVoice.git synced 2026-05-01 04:01:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

model path update, fixes

Browse Source
This commit is contained in:
WildAi
2025-09-01 11:57:35 +03:00
parent 37803a884f
commit 64fdb94e16
3 changed files with 128 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
vibevoice_nodes.py
View File

@@ -37,9 +37,9 @@ MODEL_CONFIGS = {
"size_gb": 3.0,
"tokenizer_repo": "Qwen/Qwen2.5-1.5B"
},
"VibeVoice-Large-pt": {
"repo_id": "WestZhang/VibeVoice-Large-pt",
"size_gb": 14.0,
"VibeVoice-Large": {
"repo_id": "microsoft/VibeVoice-Large",
"size_gb": 17.4,
"tokenizer_repo": "Qwen/Qwen2.5-7B"
}
}
@@ -281,14 +281,14 @@ class VibeVoiceLoader:
except Exception as e:
logger.error(f"Failed to load model with {final_attention_mode} attention: {e}")
# Progressive fallback: flash -> sdpa -> eager
if final_attention_mode == "flash_attention_2":
logger.info("Attempting fallback to SDPA...")
return VibeVoiceLoader.load_model(model_name, "sdpa")
return VibeVoiceLoader.load_model(model_name, device, "sdpa")
elif final_attention_mode == "sdpa":
logger.info("Attempting fallback to eager...")
return VibeVoiceLoader.load_model(model_name, "eager")
return VibeVoiceLoader.load_model(model_name, device, "eager")
else:
# If eager fails, something is seriously wrong
raise RuntimeError(f"Failed to load model even with eager attention: {e}")