25 lines
1.1 KiB
Python
25 lines
1.1 KiB
Python
|
|
from .nodes.comfy_nodes import EasyControlLoadFlux, EasyControlLoadLora, EasyControlLoadMultiLora, EasyControlGenerate, EasyControlLoadStyleLora, EasyControlLoadStyleLoraFromCivitai
|
|
|
|
|
|
# 注册节点
|
|
NODE_CLASS_MAPPINGS = {
|
|
"EasyControlLoadFlux": EasyControlLoadFlux,
|
|
"EasyControlLoadLora": EasyControlLoadLora,
|
|
"EasyControlLoadMultiLora": EasyControlLoadMultiLora,
|
|
"EasyControlGenerate": EasyControlGenerate,
|
|
"EasyControlLoadStyleLora": EasyControlLoadStyleLora,
|
|
"EasyControlLoadStyleLoraFromCivitai": EasyControlLoadStyleLoraFromCivitai,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"EasyControlLoadFlux": "Load EasyControl Flux Model",
|
|
"EasyControlLoadLora": "Load EasyControl Lora",
|
|
"EasyControlLoadMultiLora": "Load Multiple EasyControl Loras",
|
|
"EasyControlGenerate": "EasyControl Generate",
|
|
"EasyControlLoadStyleLora": "Load EasyControl Style Lora",
|
|
"EasyControlLoadStyleLoraFromCivitai": "Load EasyControl Style Lora from Civitai",
|
|
}
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', "WEB_DIRECTORY"] |