mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-06 22:10:14 -03:00
28 lines
732 B
Python
28 lines
732 B
Python
class LoraStackCombinerLM:
|
|
NAME = "Lora Stack Combiner (LoraManager)"
|
|
CATEGORY = "Lora Manager/stackers"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {},
|
|
"optional": {
|
|
"lora_stack_a": ("LORA_STACK",),
|
|
"lora_stack_b": ("LORA_STACK",),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("LORA_STACK",)
|
|
RETURN_NAMES = ("LORA_STACK",)
|
|
FUNCTION = "combine_stacks"
|
|
|
|
def combine_stacks(self, lora_stack_a=None, lora_stack_b=None):
|
|
combined_stack = []
|
|
|
|
if lora_stack_a:
|
|
combined_stack.extend(lora_stack_a)
|
|
if lora_stack_b:
|
|
combined_stack.extend(lora_stack_b)
|
|
|
|
return (combined_stack,)
|