From 756ad399bfa30053440ccaa0827e8ead3ac58cf9 Mon Sep 17 00:00:00 2001 From: Will Miao <13051207myq@gmail.com> Date: Sat, 15 Mar 2025 10:45:32 +0800 Subject: [PATCH] Enhance LoraManagerLoader to include formatted loaded_loras in return values, improving data output for loaded LoRAs. --- py/nodes/lora_loader.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/py/nodes/lora_loader.py b/py/nodes/lora_loader.py index ec287721..aae8d882 100644 --- a/py/nodes/lora_loader.py +++ b/py/nodes/lora_loader.py @@ -26,8 +26,8 @@ class LoraManagerLoader: "optional": FlexibleOptionalInputType(any_type), } - RETURN_TYPES = ("MODEL", "CLIP", IO.STRING) - RETURN_NAMES = ("MODEL", "CLIP", "trigger_words") + RETURN_TYPES = ("MODEL", "CLIP", IO.STRING, IO.STRING) + RETURN_NAMES = ("MODEL", "CLIP", "trigger_words", "loaded_loras") FUNCTION = "load_loras" async def get_lora_info(self, lora_name): @@ -95,5 +95,9 @@ class LoraManagerLoader: # use ',, ' to separate trigger words for group mode trigger_words_text = ",, ".join(all_trigger_words) if all_trigger_words else "" + + # Format loaded_loras as separated by spaces + formatted_loras = " ".join([f"" + for name, strength in [item.split(':') for item in loaded_loras]]) - return (model, clip, trigger_words_text) \ No newline at end of file + return (model, clip, trigger_words_text, formatted_loras) \ No newline at end of file