29 lines
1.0 KiB
Python
29 lines
1.0 KiB
Python
from comfy_api.latest import io
|
|
import folder_paths
|
|
|
|
from .control_ctrlora import load_ctrlora
|
|
|
|
class CtrLoRALoader(io.ComfyNode):
|
|
@classmethod
|
|
def define_schema(cls) -> io.Schema:
|
|
return io.Schema(
|
|
node_id='ACN_CtrLoRALoader',
|
|
display_name='Load CtrLoRA Model 🛂🅐🅒🅝',
|
|
category='Adv-ControlNet 🛂🅐🅒🅝/CtrLoRA',
|
|
inputs=[
|
|
io.Combo.Input('base', options=folder_paths.get_filename_list("controlnet")),
|
|
io.Combo.Input('lora', options=folder_paths.get_filename_list("controlnet"))
|
|
],
|
|
outputs=[
|
|
io.ControlNet.Output('CONTROL_NET', is_output_list=False)
|
|
]
|
|
)
|
|
|
|
|
|
@classmethod
|
|
def execute(cls, base: str, lora: str):
|
|
base_path = folder_paths.get_full_path("controlnet", base)
|
|
lora_path = folder_paths.get_full_path("controlnet", lora)
|
|
controlnet = load_ctrlora(base_path, lora_path)
|
|
return io.NodeOutput(controlnet,)
|