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

fix tokenizer issue

Browse Source
This commit is contained in:
WildAi
2025-09-03 20:47:05 +03:00
parent 4fe38c9bca
commit 642fd3f70a
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
vibevoice_nodes.py
View File

@@ -197,7 +197,20 @@ class VibeVoiceLoader:
logger.info(f"Loading VibeVoice model components from: {model_path}") logger.info(f"Loading VibeVoice model components from: {model_path}")
tokenizer_repo = MODEL_CONFIGS[model_name].get("tokenizer_repo") tokenizer_repo = MODEL_CONFIGS[model_name].get("tokenizer_repo")
tokenizer_file_path = hf_hub_download(repo_id=tokenizer_repo, filename="tokenizer.json") tokenizer_file_path = os.path.join(model_path, "tokenizer.json")
# Check if tokenizer.json exists locally. If not, download it directly to the model folder.
if not os.path.exists(tokenizer_file_path):
logger.info(f"tokenizer.json not found in {model_path}. Downloading from '{tokenizer_repo}'...")
try:
hf_hub_download(
repo_id=tokenizer_repo,
filename="tokenizer.json",
local_dir=model_path,
)
except Exception as e:
logger.error(f"Failed to download tokenizer.json: {e}")
raise
vibevoice_tokenizer = VibeVoiceTextTokenizerFast(tokenizer_file=tokenizer_file_path) vibevoice_tokenizer = VibeVoiceTextTokenizerFast(tokenizer_file=tokenizer_file_path)
audio_processor = VibeVoiceTokenizerProcessor() audio_processor = VibeVoiceTokenizerProcessor()