wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 21:30:15 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

WAN 2.1 support

Browse Source
This commit is contained in:
doctorpangloss 2025-03-06 07:32:04 -08:00
parent bfb145bb26
commit 83948cafd1
6 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/app/frontend_management.py
View File

@ -17,7 +17,7 @@ import importlib.resources
from typing_extensions import NotRequired from typing_extensions import NotRequired
from ..cli_args import DEFAULT_VERSION_STRING 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 REQUEST_TIMEOUT = 10 # seconds

2
comfy/model_management.py
View File

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

2
comfy/nodes/base_nodes.py
View File

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

2
comfy/nodes/package_typing.py
View File

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

1
comfy/sd.py
View File

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

4
comfy_extras/nodes/nodes_wan.py
View File

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