nodes/nkchocoai-ComfyUI-SaveImageWithMetaData
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix input_data for extension nodes

Browse Source
This commit is contained in:
Trần Nam Tuấn
2024-08-20 04:44:37 +07:00
parent d22d0adba6
commit 4ee3fe6cef
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py/defs/ext/efficiency_nodes.py
+4 -4
View File
@@ -17,22 +17,22 @@ def get_lora_model_hash_stack(node_id, obj, prompt, extra_data, outputs, input_d
def get_lora_strength_model_stack(
node_id, obj, prompt, extra_data, outputs, input_data
):
if input_data["input_mode"][0] == "advanced":
if input_data[0]["input_mode"][0] == "advanced":
return get_lora_data_stack(input_data, "model_str")
return get_lora_data_stack(input_data, "lora_wt")
def get_lora_strength_clip_stack(node_id, obj, prompt, extra_data, outputs, input_data):
if input_data["input_mode"][0] == "advanced":
if input_data[0]["input_mode"][0] == "advanced":
return get_lora_data_stack(input_data, "clip_str")
return get_lora_data_stack(input_data, "lora_wt")
def get_lora_data_stack(input_data, attribute):
lora_count = input_data["lora_count"][0]
lora_count = input_data[0]["lora_count"][0]
return [
v[0]
for k, v in input_data.items()
for k, v in input_data[0].items()
if k.startswith(attribute) and v[0] != "None"
][:lora_count]
py/defs/ext/rgthree.py
+2 -2
View File
@@ -21,7 +21,7 @@ def get_lora_strength(node_id, obj, prompt, extra_data, outputs, input_data):
def get_lora_data(input_data, attribute):
return [
v[0][attribute]
for k, v in input_data.items()
for k, v in input_data[0].items()
if k.startswith("lora_") and v[0]["on"]
]
@@ -44,7 +44,7 @@ def get_lora_strength_stack(node_id, obj, prompt, extra_data, outputs, input_dat
def get_lora_data_stack(input_data, attribute):
return [
v[0]
for k, v in input_data.items()
for k, v in input_data[0].items()
if k.startswith(attribute + "_") and v[0] != "None"
]