Merge pull request #21 from ComfyNodePRs/update-publish-yaml

Update Github Action for Publishing to Comfy Registry

.github/workflows/publish.yml

@@ -7,15 +7,19 @@ on:
     paths:
       - "pyproject.toml"
 
+permissions:
+  issues: write
+
 jobs:
   publish-node:
     name: Publish Custom Node to registry
     runs-on: ubuntu-latest
+    if: ${{ github.repository_owner == '54rt1n' }}
     steps:
       - name: Check out code
         uses: actions/checkout@v4
       - name: Publish Custom Node
-        uses: Comfy-Org/publish-node-action@main
+        uses: Comfy-Org/publish-node-action@v1
         with:
           ## Add your own personal access token to your Github Repository secrets and reference it here.
           personal_access_token: ${{ secrets.REGISTRY_ACCESS_TOKEN }}

ddare/tensor.py

@@ -54,7 +54,7 @@ def dare_ties_sparsification(model_a_param: torch.Tensor, model_b_param: torch.T
     model_b_flat = model_b_param.view(-1).float().to(device)
     delta_flat = model_b_flat - model_a_flat
 
-    dare_mask = torch.bernoulli(torch.full(delta_flat.shape, 1 - drop_rate, device=device)).bool()
+    dare_mask = torch.bernoulli(torch.full(delta_flat.shape, 1.0 - drop_rate, device=device)).bool()
     # The paper says we should rescale, but it yields terrible results for SD.
     if rescale == "on":
         # Rescale the remaining deltas
@@ -236,4 +236,4 @@ def divide_tensor_into_sets(tensor : torch.Tensor, n_sets : int):
     # Randomly assign each element to one of the n sets
     random_set_assignments = torch.randint(0, n_sets, tensor.shape, dtype=torch.int)
 
-    return random_set_assignments
+    return random_set_assignments

pyproject.toml

@@ -1,8 +1,8 @@
 [project]
 name = "comfyui-daremerge"
 description = "ComfyUI powertools for SD1.5 and SDXL model merging"
-version = "0.3.1"
-license = "GPL-3.0-only"
+version = "0.3.2"
+license = { text = "GNU General Public License v3.0" }
 authors = [
     { name = "54rt1n", email = "opensource@bmds.us" }
 ]