nodes/nobandegani-ComfyUI-InoNodes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update init_helper.py

Browse Source
This commit is contained in:
Inoland
2025-12-27 18:26:23 +04:00
parent 8181a0bc46
commit 94cdede641
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/comfyui_ino_nodes/init_helper.py
+14 -6
View File
@@ -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.")