alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-26 15:38:52 -03:00
Code Issues Packages Projects Releases Wiki Activity

Enhance LoraStacker to return active LoRAs in stack_loras method

Browse Source 
- Updated RETURN_TYPES and RETURN_NAMES to include active LoRAs.
- Introduced active_loras list to track active LoRAs and their strengths.
- Formatted active_loras for return as a string in the format <lora:lora_name:strength>.
This commit is contained in:
Will Miao
2025-03-27 16:10:50 +08:00
parent ba18cbabfd
commit 6e3b4178ac
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
py/nodes/lora_stacker.py
View File

@@ -26,8 +26,8 @@ class LoraStacker:
"optional": FlexibleOptionalInputType(any_type), "optional": FlexibleOptionalInputType(any_type),
} }
RETURN_TYPES = ("LORA_STACK", IO.STRING) RETURN_TYPES = ("LORA_STACK", IO.STRING, IO.STRING)
RETURN_NAMES = ("LORA_STACK", "trigger_words") RETURN_NAMES = ("LORA_STACK", "trigger_words", "active_loras")
FUNCTION = "stack_loras" FUNCTION = "stack_loras"
async def get_lora_info(self, lora_name): async def get_lora_info(self, lora_name):
@@ -75,6 +75,7 @@ class LoraStacker:
def stack_loras(self, text, **kwargs): def stack_loras(self, text, **kwargs):
"""Stacks multiple LoRAs based on the kwargs input without loading them.""" """Stacks multiple LoRAs based on the kwargs input without loading them."""
stack = [] stack = []
active_loras = []
all_trigger_words = [] all_trigger_words = []
# Process existing lora_stack if available # Process existing lora_stack if available
@@ -103,11 +104,15 @@ class LoraStacker:
# Add to stack without loading # Add to stack without loading
# replace '/' with os.sep to avoid different OS path format # replace '/' with os.sep to avoid different OS path format
stack.append((lora_path.replace('/', os.sep), model_strength, clip_strength)) stack.append((lora_path.replace('/', os.sep), model_strength, clip_strength))
active_loras.append((lora_name, model_strength))
# Add trigger words to collection # Add trigger words to collection
all_trigger_words.extend(trigger_words) all_trigger_words.extend(trigger_words)
# use ',, ' to separate trigger words for group mode # use ',, ' to separate trigger words for group mode
trigger_words_text = ",, ".join(all_trigger_words) if all_trigger_words else "" trigger_words_text = ",, ".join(all_trigger_words) if all_trigger_words else ""
# Format active_loras as <lora:lora_name:strength> separated by spaces
active_loras_text = " ".join([f"<lora:{name}:{str(strength).strip()}>"
for name, strength in active_loras])
return (stack, trigger_words_text) return (stack, trigger_words_text, active_loras_text)