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