From 4ee3fe6cefd954cdceb0e72a7cae3ecc7b516923 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tr=E1=BA=A7n=20Nam=20Tu=E1=BA=A5n?= Date: Tue, 20 Aug 2024 04:44:37 +0700 Subject: [PATCH] Fix input_data for extension nodes --- py/defs/ext/efficiency_nodes.py | 8 ++++---- py/defs/ext/rgthree.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/py/defs/ext/efficiency_nodes.py b/py/defs/ext/efficiency_nodes.py index bf15b86..d341ee7 100644 --- a/py/defs/ext/efficiency_nodes.py +++ b/py/defs/ext/efficiency_nodes.py @@ -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] diff --git a/py/defs/ext/rgthree.py b/py/defs/ext/rgthree.py index aad025f..887ecd2 100644 --- a/py/defs/ext/rgthree.py +++ b/py/defs/ext/rgthree.py @@ -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" ]