From 94cdede64149fd8ad2202034b0e9a9cd0f5dc1e1 Mon Sep 17 00:00:00 2001 From: Inoland Date: Sat, 27 Dec 2025 18:26:23 +0400 Subject: [PATCH] Update init_helper.py --- src/comfyui_ino_nodes/init_helper.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/src/comfyui_ino_nodes/init_helper.py b/src/comfyui_ino_nodes/init_helper.py index 0e8e8d1..84bddd3 100644 --- a/src/comfyui_ino_nodes/init_helper.py +++ b/src/comfyui_ino_nodes/init_helper.py @@ -13,22 +13,25 @@ async def init_models(): huggingface_downloader = InoHuggingFaceDownloadModel() - wan_animate_model_bf16 = await handle_download_model.function( + _, wan_animate_model_bf16, _ = await handle_download_model.function( enabled=True, config=video_model_config_loader.function(True, "comfy-Wan-22-Animate-14B-bf16"), ) + print(f"wan_animate_model_bf16: {wan_animate_model_bf16}") - wan_22_clip_bf16 = await handle_download_model.function( + _, wan_22_clip_bf16, _ = await handle_download_model.function( enabled=True, config=clip_model_config_loader.function(True, "kj-umt5-bf16"), ) + print(f"wan_22_clip_bf16: {wan_22_clip_bf16}") - wan_21_vae_bf16 = await handle_download_model.function( + _, wan_21_vae_bf16, _ = await handle_download_model.function( enabled=True, config=vae_model_config_loader.function(True, "kj-wan-21-bf16"), ) + print(f"wan_21_vae_bf16: {wan_21_vae_bf16}") - wan_animate_yolo = await huggingface_downloader.function( + _, wan_animate_yolo, _ = await huggingface_downloader.function( enabled=True, model_config="{}", model_type="detection", @@ -36,7 +39,9 @@ async def init_models(): repo_id="Wan-AI/Wan2.2-Animate-14B", filename="process_checkpoint/det/yolov10m.onnx", ) - wan_animate_vitpose_bin = await huggingface_downloader.function( + print(f"wan_animate_yolo: {wan_animate_yolo}") + + _, wan_animate_vitpose_bin, _ = await huggingface_downloader.function( enabled=True, model_config="{}", model_type="detection", @@ -44,7 +49,9 @@ async def init_models(): repo_id="Kijai/vitpose_comfy", filename="onnx/vitpose_h_wholebody_data.bin", ) - wan_animate_vitpose_onnx = await huggingface_downloader.function( + print(f"wan_animate_vitpose_bin: {wan_animate_vitpose_bin}") + + _, wan_animate_vitpose_onnx, _ = await huggingface_downloader.function( enabled=True, model_config="{}", model_type="detection", @@ -52,5 +59,6 @@ async def init_models(): repo_id="Kijai/vitpose_comfy", filename="onnx/vitpose_h_wholebody_model.onnx", ) + print(f"wan_animate_vitpose_onnx: {wan_animate_vitpose_onnx}") print("Init models finished.")