Merge pull request #47 from Hapseleg/master

Added compatibility with ComfyUI-Easy-Use sampler and loader
2024-10-27 14:44:41 +09:00
parent 200d209039 07f4b292fa
commit 1a416be029
1 changed files with 93 additions and 0 deletions
@@ -0,0 +1,93 @@
+# https://github.com/yolain/ComfyUI-Easy-Use
+from ..meta import MetaField
+from ..formatters import calc_model_hash, calc_lora_hash, convert_skip_clip
+import re
+
+def get_lora_model_name_stack(node_id, obj, prompt, extra_data, outputs, input_data):
+    return get_lora_data_stack(input_data, "lora_\d_name")
+
+
+def get_lora_model_hash_stack(node_id, obj, prompt, extra_data, outputs, input_data):
+    return [
+        calc_lora_hash(model_name, input_data)
+        for model_name in get_lora_data_stack(input_data, "lora_\d_name")
+    ]
+
+
+def get_lora_strength_model_stack(node_id, obj, prompt, extra_data, outputs, input_data):
+    if input_data[0]["mode"][0] == "advanced":
+        return get_lora_data_stack(input_data, "lora_\d_model_strength")
+    return get_lora_data_stack(input_data, "lora_\d_strength")
+
+
+def get_lora_strength_clip_stack(node_id, obj, prompt, extra_data, outputs, input_data):
+    if input_data[0]["mode"][0] == "advanced":
+        return get_lora_data_stack(input_data, "lora_\d_clip_strength")
+    return get_lora_data_stack(input_data, "lora_\d_strength")
+
+
+def get_lora_data_stack(input_data, attribute):
+    lora_count = input_data[0]["num_loras"][0]
+    return [
+        v[0]
+        for k, v in input_data[0].items()
+        if re.search(attribute, k) != None and v[0] != "None"
+    ][:lora_count]
+
+
+def get_lora_model_hash(node_id, obj, prompt, extra_data, outputs, input_data):
+    if input_data[0]["lora_name"][0] != "None":
+        return calc_lora_hash(input_data[0]["lora_name"][0], input_data)
+    else:
+        return ""
+
+
+SAMPLERS = {
+    "easy fullkSampler": {
+        "positive": "positive",
+        "negative": "negative",
+    },
+}
+
+
+CAPTURE_FIELD_LIST = {
+    "easy fullLoader": {
+        MetaField.MODEL_NAME: {"field_name": "ckpt_name"},
+        MetaField.MODEL_HASH: {"field_name": "ckpt_name", "format": calc_model_hash},
+        MetaField.CLIP_SKIP: {"field_name": "clip_skip", "format": convert_skip_clip},
+        MetaField.POSITIVE_PROMPT: {"field_name": "positive"},
+        MetaField.NEGATIVE_PROMPT: {"field_name": "negative"},
+        MetaField.IMAGE_WIDTH: {"field_name": "empty_latent_width"},
+        MetaField.IMAGE_HEIGHT: {"field_name": "empty_latent_height"},
+        MetaField.LORA_MODEL_NAME: {"field_name": "lora_name"},
+        MetaField.LORA_MODEL_HASH: {"selector": get_lora_model_hash},
+        MetaField.LORA_STRENGTH_MODEL: {"field_name": "lora_model_strength"},
+        MetaField.LORA_STRENGTH_CLIP: {"field_name": "lora_clip_strength"},
+    },
+    "easy comfyLoader": {
+        MetaField.MODEL_NAME: {"field_name": "ckpt_name"},
+        MetaField.MODEL_HASH: {"field_name": "ckpt_name", "format": calc_model_hash},
+        MetaField.CLIP_SKIP: {"field_name": "clip_skip", "format": convert_skip_clip},
+        MetaField.POSITIVE_PROMPT: {"field_name": "positive"},
+        MetaField.NEGATIVE_PROMPT: {"field_name": "negative"},
+        MetaField.IMAGE_WIDTH: {"field_name": "empty_latent_width"},
+        MetaField.IMAGE_HEIGHT: {"field_name": "empty_latent_height"},
+        MetaField.LORA_MODEL_NAME: {"field_name": "lora_name"},
+        MetaField.LORA_MODEL_HASH: {"selector": get_lora_model_hash},
+        MetaField.LORA_STRENGTH_MODEL: {"field_name": "lora_model_strength"},
+        MetaField.LORA_STRENGTH_CLIP: {"field_name": "lora_clip_strength"},
+    },
+    "easy fullkSampler": {
+        MetaField.SEED: {"field_name": "seed"},
+        MetaField.STEPS: {"field_name": "steps"},
+        MetaField.CFG: {"field_name": "cfg"},
+        MetaField.SAMPLER_NAME: {"field_name": "sampler_name"},
+        MetaField.SCHEDULER: {"field_name": "scheduler"},
+    },
+    "easy loraStack": {
+        MetaField.LORA_MODEL_NAME: {"selector": get_lora_model_name_stack},
+        MetaField.LORA_MODEL_HASH: {"selector": get_lora_model_hash_stack},
+        MetaField.LORA_STRENGTH_MODEL: {"selector": get_lora_strength_model_stack},
+        MetaField.LORA_STRENGTH_CLIP: {"selector": get_lora_strength_clip_stack},
+    },
+}