Update init_helper.py

2025-12-27 18:16:17 +04:00
parent 77e328b8ad
commit 3f57470da4
1 changed files with 22 additions and 1 deletions
@@ -1,12 +1,33 @@
 import asyncio

-from .workflow_helpers.download_model_helper import InoHuggingFaceDownloadModel
+from .workflow_helpers.download_model_helper import InoHuggingFaceDownloadModel, InoGetVideoModelDownloadConfig, InoGetClipDownloadConfig, InoGetVaeDownloadConfig, InoHandleDownloadModel

 async def init_models():
    print("Init models started.")

+    video_model_config_loader = InoGetVideoModelDownloadConfig()
+    clip_model_config_loader = InoGetClipDownloadConfig()
+    vae_model_config_loader = InoGetVaeDownloadConfig()
+
+    handle_download_model = InoHandleDownloadModel()
+
    huggingface_downloader = InoHuggingFaceDownloadModel()

+    wan_animate_model_bf16 = handle_download_model.function(
+        enabled=True,
+        config=video_model_config_loader.function(True, "comfy-Wan-22-Animate-14B-bf16"),
+    )
+
+    wan_22_clip_bf16 = handle_download_model.function(
+        enabled=True,
+        config=clip_model_config_loader.function(True, "kj-umt5-bf16"),
+    )
+
+    wan_21_vae_bf16 = handle_download_model.function(
+        enabled=True,
+        config=vae_model_config_loader.function(True, "kj-wan-21-bf16"),
+    )
+
    wan_animate_yolo = await huggingface_downloader.function(
        enabled=True,
        model_config="{}",