WAN 2.1 support

comfy/app/frontend_management.py

@@ -17,7 +17,7 @@ import importlib.resources
 from typing_extensions import NotRequired
 
 from ..cli_args import DEFAULT_VERSION_STRING
-from ..cmd.folder_paths import add_model_folder_path
+from ..cmd.folder_paths import add_model_folder_path  # pylint: disable=import-error
 
 REQUEST_TIMEOUT = 10  # seconds
 

comfy/model_management.py

@@ -120,7 +120,7 @@ except:
     npu_available = False
 
 try:
-    import torch_mlu  # noqa: F401
+    import torch_mlu  # pylint: disable=import-error, noqa: F401
 
     _ = torch.mlu.device_count()
     mlu_available = torch.mlu.is_available()

comfy/nodes/base_nodes.py

@@ -977,7 +977,7 @@ class CLIPLoader:
         elif type == "lumina2":
             clip_type = sd.CLIPType.LUMINA2
         elif type == "wan":
-            clip_type = comfy.sd.CLIPType.WAN
+            clip_type = sd.CLIPType.WAN
         else:
             logging.warning(f"Unknown clip type argument passed: {type} for model {clip_name}")
 

comfy/nodes/package_typing.py

@@ -101,7 +101,7 @@ class SaveNodeResultT(TypedDict, total=True):
     type: Literal["output", "input", "temp"]
 
 
-SaveNodeResult = SaveNodeResultT | FileLocator
+SaveNodeResult = SaveNodeResultT
 
 
 class UIImagesImagesResult(TypedDict, total=True):

comfy/sd.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import dataclasses
+import json
 import logging
 import math
 import os.path

comfy_extras/nodes/nodes_wan.py

@@ -1,5 +1,5 @@
-import nodes
-import node_helpers
+from comfy.nodes import base_nodes as nodes
+from comfy import node_helpers
 import torch
 import comfy.model_management
 import comfy.utils