From f78ed63b678952cfb2ea2d2dd15bb49e74dd262a Mon Sep 17 00:00:00 2001 From: jitcoder Date: Fri, 6 Sep 2024 12:47:10 +0500 Subject: [PATCH] feat: added lora name as output --- lora_info.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lora_info.py b/lora_info.py index 58b862f..e73d8d2 100644 --- a/lora_info.py +++ b/lora_info.py @@ -128,14 +128,14 @@ class LoraInfo: }, } - RETURN_NAMES = ("trigger_words", "example_prompt") - RETURN_TYPES = ("STRING", "STRING") + RETURN_NAMES = ("lora_name", "trigger_words", "example_prompt") + RETURN_TYPES = ("STRING", "STRING", "STRING") FUNCTION = "lora_info" OUTPUT_NODE = True CATEGORY = "jitcoder" def lora_info(self, lora_name): (output, triggerWords, examplePrompt, baseModel) = get_lora_info(lora_name) - return {"ui": {"text": (output,), "model": (baseModel,)}, "result": (triggerWords, examplePrompt)} + return {"ui": {"text": (output,), "model": (baseModel,)}, "result": (lora_name, triggerWords, examplePrompt)}