mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from .py.lora_manager import LoraManager
|
|
from .py.nodes.lora_loader import LoraManagerLoader, LoraManagerTextLoader
|
|
from .py.nodes.trigger_word_toggle import TriggerWordToggle
|
|
from .py.nodes.lora_stacker import LoraStacker
|
|
from .py.nodes.save_image import SaveImage
|
|
from .py.nodes.debug_metadata import DebugMetadata
|
|
from .py.nodes.wanvideo_lora_select import WanVideoLoraSelect
|
|
from .py.nodes.wanvideo_lora_select_from_text import WanVideoLoraSelectFromText
|
|
# Import metadata collector to install hooks on startup
|
|
from .py.metadata_collector import init as init_metadata_collector
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
LoraManagerLoader.NAME: LoraManagerLoader,
|
|
LoraManagerTextLoader.NAME: LoraManagerTextLoader,
|
|
TriggerWordToggle.NAME: TriggerWordToggle,
|
|
LoraStacker.NAME: LoraStacker,
|
|
SaveImage.NAME: SaveImage,
|
|
DebugMetadata.NAME: DebugMetadata,
|
|
WanVideoLoraSelect.NAME: WanVideoLoraSelect,
|
|
WanVideoLoraSelectFromText.NAME: WanVideoLoraSelectFromText
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web/comfyui"
|
|
|
|
# Initialize metadata collector
|
|
init_metadata_collector()
|
|
|
|
# Register routes on import
|
|
LoraManager.add_routes()
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'WEB_DIRECTORY']
|