nodes/jitcoder-lora-info
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: added lora name as output

Browse Source
This commit is contained in:
jitcoder
2024-09-06 12:47:10 +05:00
parent 97bdc494d7
commit f78ed63b67
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lora_info.py
+3 -3
View File
@@ -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)}