diff --git a/README.md b/README.md index a23775a..4fe951f 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# ComfyUI-Nunchaku QwenImage&Z-ImageTurboLoraLoader and Diffsynth ControlNet Loader +# ComfyUI-Nunchaku QwenImage&UniversalLoraStack and Diffsynth ControlNet Loader A ComfyUI custom node for loading and applying LoRA (Low-Rank Adaptation) to Nunchaku Qwen Image and Z-ImageTurbo models, and diffsynth ControlNet functionality. ComfyUI Nodes 2.0 compatible. diff --git a/__init__.py b/__init__.py index 498875f..4d9165d 100644 --- a/__init__.py +++ b/__init__.py @@ -23,7 +23,9 @@ try: from .nodes.lora.qwenimage_v3 import GENERATED_NODES as QWEN_V3_NODES, GENERATED_DISPLAY_NAMES as QWEN_V3_NAMES # Z-Image-Turbo V2 is deprecated (unofficial loader only) - removed from registration to avoid confusion # from .nodes.lora.zimageturbo_v2 import GENERATED_NODES as ZIMAGETURBO_V2_NODES, GENERATED_DISPLAY_NAMES as ZIMAGETURBO_V2_NAMES - from .nodes.lora.zimageturbo_v3 import GENERATED_NODES as ZIMAGETURBO_V3_NODES, GENERATED_DISPLAY_NAMES as ZIMAGETURBO_V3_NAMES + # Z-Image-Turbo V3 is deprecated - removed from registration + # from .nodes.lora.zimageturbo_v3 import GENERATED_NODES as ZIMAGETURBO_V3_NODES, GENERATED_DISPLAY_NAMES as ZIMAGETURBO_V3_NAMES + from .nodes.lora.zimageturbo_v4 import GENERATED_NODES as ZIMAGETURBO_V4_NODES, GENERATED_DISPLAY_NAMES as ZIMAGETURBO_V4_NAMES # Add version to classes before creating NODE_CLASS_MAPPINGS NunchakuQwenImageLoraLoader.__version__ = __version__ @@ -35,7 +37,10 @@ try: # Z-Image-Turbo V2 registration removed (unofficial loader only) # for node_class in ZIMAGETURBO_V2_NODES.values(): # node_class.__version__ = __version__ - for node_class in ZIMAGETURBO_V3_NODES.values(): + # Z-Image-Turbo V3 registration removed + # for node_class in ZIMAGETURBO_V3_NODES.values(): + # node_class.__version__ = __version__ + for node_class in ZIMAGETURBO_V4_NODES.values(): node_class.__version__ = __version__ NODE_CLASS_MAPPINGS["NunchakuQwenImageLoraLoader"] = NunchakuQwenImageLoraLoader @@ -44,7 +49,9 @@ try: NODE_CLASS_MAPPINGS.update(QWEN_V3_NODES) # Z-Image-Turbo V2 registration removed (unofficial loader only) # NODE_CLASS_MAPPINGS.update(ZIMAGETURBO_V2_NODES) - NODE_CLASS_MAPPINGS.update(ZIMAGETURBO_V3_NODES) + # Z-Image-Turbo V3 registration removed + # NODE_CLASS_MAPPINGS.update(ZIMAGETURBO_V3_NODES) + NODE_CLASS_MAPPINGS.update(ZIMAGETURBO_V4_NODES) except ImportError: logger.exception("LoRA nodes import failed:") @@ -66,7 +73,9 @@ NODE_DISPLAY_NAME_MAPPINGS = { **QWEN_V3_NAMES, # Z-Image-Turbo V2 registration removed (unofficial loader only) # **ZIMAGETURBO_V2_NAMES, - **ZIMAGETURBO_V3_NAMES + # Z-Image-Turbo V3 registration removed + # **ZIMAGETURBO_V3_NAMES, + **ZIMAGETURBO_V4_NAMES } # Add ControlNet display name only if the node was successfully loaded