Merge pull request #5 from ComfyNodePRs/publish

Add Github Action for Publishing to Comfy Registry

.github/workflows/publish.yml

@@ -0,0 +1,20 @@
+name: Publish to Comfy registry
+on:
+  workflow_dispatch:
+  push:
+    tags:
+      - "[0-9]+.[0-9]+.[0-9]+"
+    paths:
+      - "pyproject.toml"
+
+jobs:
+  publish-node:
+    name: Publish Custom Node to registry
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v4
+      - name: Publish Custom Node
+        uses: Comfy-Org/publish-node-action@main
+        with:
+          personal_access_token: ${{ secrets.REGISTRY_ACCESS_TOKEN }}

attention_couple.py

@@ -116,8 +116,8 @@ class AttentionCouple:
             masks_uncond = get_masks_from_q(self.negative_positive_masks[0], q_list[0], extra_options["original_shape"])
             masks_cond = get_masks_from_q(self.negative_positive_masks[1], q_list[0], extra_options["original_shape"])
 
-            context_uncond = torch.cat([cond for cond in self.negative_positive_conds[0]], dim=0)
-            context_cond = torch.cat([cond for cond in self.negative_positive_conds[1]], dim=0)
+            context_uncond = self.negative_positive_conds[0][0]
+            context_cond = self.negative_positive_conds[1][0]
             
             k_uncond = module.to_k(context_uncond)
             k_cond = module.to_k(context_cond)
@@ -138,8 +138,6 @@ class AttentionCouple:
                     length = len_neg
 
                 q_target = q_list[i].repeat(length, 1, 1)
-                k = torch.cat([k[i].unsqueeze(0).repeat(b,1,1) for i in range(length)], dim=0)
-                v = torch.cat([v[i].unsqueeze(0).repeat(b,1,1) for i in range(length)], dim=0)
 
                 qkv = optimized_attention(q_target, k, v, extra_options["n_heads"])
                 qkv = qkv * masks

pyproject.toml

@@ -0,0 +1,13 @@
+[project]
+name = "comfyui-comfycouple"
+description = "If you want to draw two different characters together without blending their features, so you could try to check out this custom node."
+version = "0.1.2"
+license = "GPL-3.0"
+
+[project.urls]
+Repository = "https://github.com/Danand/ComfyUI-ComfyCouple"
+
+[tool.comfy]
+PublisherId = "rei---"
+DisplayName = "ComfyUI-ComfyCouple"
+Icon = ""