Don't patch weights when multiplier is zero.

6ad0a6d7 · comfyanonymous · af15add9 · 6ad0a6d7
Commit 6ad0a6d7 authored Jul 09, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 4 deletions

comfy/sd.py comfy/sd.py +5 -4

No files found.
--- a/comfy/sd.py
+++ b/comfy/sd.py
@@ -359,10 +359,11 @@ class ModelPatcher:

            if len(v) == 1:
                w1 = v[0]
-                if w1.shape != weight.shape:
-                    print("WARNING SHAPE MISMATCH {} WEIGHT NOT MERGED {} != {}".format(key, w1.shape, weight.shape))
-                else:
-                    weight += alpha * w1.type(weight.dtype).to(weight.device)
+                if alpha != 0.0:
+                    if w1.shape != weight.shape:
+                        print("WARNING SHAPE MISMATCH {} WEIGHT NOT MERGED {} != {}".format(key, w1.shape, weight.shape))
+                    else:
+                        weight += alpha * w1.type(weight.dtype).to(weight.device)
            elif len(v) == 4: #lora/locon
                mat1 = v[0]
                mat2 = v[1]