Support efficiency-nodes

2024-02-17 22:34:27 +09:00
parent d636bf2c07
commit fb6afeaaa9
3 changed files with 782 additions and 0 deletions
@@ -0,0 +1,61 @@
+# https://github.com/jags111/efficiency-nodes-comfyui
+from ..meta import MetaField
+from ..formatters import calc_model_hash
+
+SAMPLERS = {
+    "KSampler (Efficient)": {
+        "positive": "positive",
+        "negative": "negative",
+    },
+    "KSampler Adv. (Efficient)": {
+        "positive": "positive",
+        "negative": "negative",
+    },
+    "KSampler SDXL (Eff.)": {
+        "positive": "positive",
+        "negative": "negative",
+    },
+}
+
+CAPTURE_FIELD_LIST = {
+    "Efficient Loader": {
+        MetaField.MODEL_NAME: {"field_name": "ckpt_name"},
+        MetaField.MODEL_HASH: {"field_name": "ckpt_name", "format": calc_model_hash},
+        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"},
+    },
+    "Eff. Loader SDXL": {
+        MetaField.MODEL_NAME: {"field_name": "base_ckpt_name"},
+        MetaField.MODEL_HASH: {
+            "field_name": "base_ckpt_name",
+            "format": calc_model_hash,
+        },
+        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"},
+    },
+    "KSampler (Efficient)": {
+        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"},
+    },
+    "KSampler Adv. (Efficient)": {
+        MetaField.SEED: {"field_name": "noise_seed"},
+        MetaField.STEPS: {"field_name": "steps"},
+        MetaField.CFG: {"field_name": "cfg"},
+        MetaField.SAMPLER_NAME: {"field_name": "sampler_name"},
+        MetaField.SCHEDULER: {"field_name": "scheduler"},
+    },
+    "KSampler SDXL (Eff.)": {
+        MetaField.SEED: {"field_name": "noise_seed"},
+        MetaField.STEPS: {"field_name": "steps"},
+        MetaField.CFG: {"field_name": "cfg"},
+        MetaField.SAMPLER_NAME: {"field_name": "sampler_name"},
+        MetaField.SCHEDULER: {"field_name": "scheduler"},
+    },
+}