From 24ba086e0df1ccd843806dc0024c19838ced3050 Mon Sep 17 00:00:00 2001 From: Will Miao <13051207myq@gmail.com> Date: Mon, 24 Feb 2025 09:58:08 +0800 Subject: [PATCH] Fix node register conflict --- __init__.py | 8 ++------ nodes/lora_loader.py | 2 +- 2 files changed, 3 insertions(+), 7 deletions(-) diff --git a/__init__.py b/__init__.py index d9083441..4e6d9ea3 100644 --- a/__init__.py +++ b/__init__.py @@ -2,15 +2,11 @@ from .lora_manager import LoraManager from .nodes.lora_loader import LoraManagerLoader NODE_CLASS_MAPPINGS = { - "LoRALoader": LoraManagerLoader -} - -NODE_DISPLAY_NAME_MAPPINGS = { - "LoRALoader": "Lora Loader (LoraManager)" + LoraManagerLoader.NAME: LoraManagerLoader } WEB_DIRECTORY = "./web/comfyui" # Register routes on import LoraManager.add_routes() -__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY'] \ No newline at end of file +__all__ = ['NODE_CLASS_MAPPINGS', 'WEB_DIRECTORY'] \ No newline at end of file diff --git a/nodes/lora_loader.py b/nodes/lora_loader.py index 0c06ed0b..0a220e6b 100644 --- a/nodes/lora_loader.py +++ b/nodes/lora_loader.py @@ -7,7 +7,7 @@ import asyncio import os class LoraManagerLoader: - NAME = "Lora Loader" + NAME = "Lora Loader (LoraManager)" CATEGORY = "loaders" @classmethod