Add AttentionCouple node

fofr · Jul 15, 2024 · 2bdc76d · 2bdc76d
1 parent fe1b20d
commit 2bdc76d
Show file tree

Hide file tree

Showing 4 changed files with 1,146 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,7 @@
 - Add [SUPIR](https://github.com/kijai/ComfyUI-SUPIR) custom nodes
 - Add KJ's [IC Light](https://github.com/kijai/ComfyUI-IC-Light) custom nodes
 - Add [StyleAligned](https://github.com/brianfitzgerald/style_aligned_comfy) custom nodes
+- Add [AttentionCouple](https://github.com/laksjdjf/attention-couple-ComfyUI) custom nodes
 - TensorRT engines added for SD3 on A40, A100 and H100
 - ComfyUI enums helper added for easy access to schedulers and samplers
 - Update [ReActor](https://github.com/Gourieff/comfyui-reactor-node/compare/51ae8f6...bcfd45c) to latest

diff --git a/custom_nodes.json b/custom_nodes.json
@@ -150,5 +150,9 @@
   {
     "repo": "https://github.com/brianfitzgerald/style_aligned_comfy",
     "commit": "39741e3"
+  },
+  {
+    "repo": "https://github.com/laksjdjf/attention-couple-ComfyUI",
+    "commit": "8730461"
   }
 ]
diff --git a/examples/api_workflows/attention_couple_api.json b/examples/api_workflows/attention_couple_api.json
@@ -0,0 +1,253 @@
+{
+  "3": {
+    "inputs": {
+      "seed": 4545,
+      "steps": 20,
+      "cfg": 8,
+      "sampler_name": "euler",
+      "scheduler": "normal",
+      "denoise": 1,
+      "model": [
+        "18",
+        0
+      ],
+      "positive": [
+        "18",
+        1
+      ],
+      "negative": [
+        "18",
+        2
+      ],
+      "latent_image": [
+        "5",
+        0
+      ]
+    },
+    "class_type": "KSampler",
+    "_meta": {
+      "title": "KSampler"
+    }
+  },
+  "4": {
+    "inputs": {
+      "ckpt_name": "juggernautXL_juggernautX.safetensors"
+    },
+    "class_type": "CheckpointLoaderSimple",
+    "_meta": {
+      "title": "Load Checkpoint"
+    }
+  },
+  "5": {
+    "inputs": {
+      "width": 768,
+      "height": 768,
+      "batch_size": 1
+    },
+    "class_type": "EmptyLatentImage",
+    "_meta": {
+      "title": "Empty Latent Image"
+    }
+  },
+  "6": {
+    "inputs": {
+      "text": "a landscape at night",
+      "clip": [
+        "4",
+        1
+      ]
+    },
+    "class_type": "CLIPTextEncode",
+    "_meta": {
+      "title": "CLIP Text Encode (Prompt)"
+    }
+  },
+  "7": {
+    "inputs": {
+      "text": "realistic, real life, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name",
+      "clip": [
+        "4",
+        1
+      ]
+    },
+    "class_type": "CLIPTextEncode",
+    "_meta": {
+      "title": "CLIP Text Encode (Prompt)"
+    }
+  },
+  "8": {
+    "inputs": {
+      "samples": [
+        "3",
+        0
+      ],
+      "vae": [
+        "4",
+        2
+      ]
+    },
+    "class_type": "VAEDecode",
+    "_meta": {
+      "title": "VAE Decode"
+    }
+  },
+  "9": {
+    "inputs": {
+      "filename_prefix": "ComfyUI",
+      "images": [
+        "8",
+        0
+      ]
+    },
+    "class_type": "SaveImage",
+    "_meta": {
+      "title": "Save Image"
+    }
+  },
+  "10": {
+    "inputs": {
+      "text": "a landscape at midday",
+      "clip": [
+        "4",
+        1
+      ]
+    },
+    "class_type": "CLIPTextEncode",
+    "_meta": {
+      "title": "CLIP Text Encode (Prompt)"
+    }
+  },
+  "11": {
+    "inputs": {
+      "x": 0,
+      "y": 0,
+      "operation": "add",
+      "destination": [
+        "13",
+        0
+      ],
+      "source": [
+        "12",
+        0
+      ]
+    },
+    "class_type": "MaskComposite",
+    "_meta": {
+      "title": "MaskComposite"
+    }
+  },
+  "12": {
+    "inputs": {
+      "value": 1,
+      "width": 384,
+      "height": 768
+    },
+    "class_type": "SolidMask",
+    "_meta": {
+      "title": "SolidMask"
+    }
+  },
+  "13": {
+    "inputs": {
+      "value": 0,
+      "width": 768,
+      "height": 768
+    },
+    "class_type": "SolidMask",
+    "_meta": {
+      "title": "SolidMask"
+    }
+  },
+  "14": {
+    "inputs": {
+      "x": 384,
+      "y": 0,
+      "operation": "add",
+      "destination": [
+        "13",
+        0
+      ],
+      "source": [
+        "12",
+        0
+      ]
+    },
+    "class_type": "MaskComposite",
+    "_meta": {
+      "title": "MaskComposite"
+    }
+  },
+  "15": {
+    "inputs": {
+      "strength": 1,
+      "set_cond_area": "default",
+      "conditioning": [
+        "10",
+        0
+      ],
+      "mask": [
+        "14",
+        0
+      ]
+    },
+    "class_type": "ConditioningSetMask",
+    "_meta": {
+      "title": "Conditioning (Set Mask)"
+    }
+  },
+  "16": {
+    "inputs": {
+      "strength": 1,
+      "set_cond_area": "default",
+      "conditioning": [
+        "6",
+        0
+      ],
+      "mask": [
+        "11",
+        0
+      ]
+    },
+    "class_type": "ConditioningSetMask",
+    "_meta": {
+      "title": "Conditioning (Set Mask)"
+    }
+  },
+  "17": {
+    "inputs": {
+      "conditioning_1": [
+        "16",
+        0
+      ],
+      "conditioning_2": [
+        "15",
+        0
+      ]
+    },
+    "class_type": "ConditioningCombine",
+    "_meta": {
+      "title": "Conditioning (Combine)"
+    }
+  },
+  "18": {
+    "inputs": {
+      "mode": "Attention",
+      "model": [
+        "4",
+        0
+      ],
+      "positive": [
+        "17",
+        0
+      ],
+      "negative": [
+        "7",
+        0
+      ]
+    },
+    "class_type": "Attention couple",
+    "_meta": {
+      "title": "Load Attention couple"
+    }
+  }
+}