Don't merge weights when shapes don't match and print a warning.

2 years ago · 9f83b098c9
parent f87ec10a97
commit 9f83b098c9
1 changed files with 5 additions and 1 deletions
--- a/comfy/sd.py
+++ b/comfy/sd.py
@ -390,7 +390,11 @@ class ModelPatcher:
                    weight *= strength_model

                if len(v) == 1:
-                    weight += alpha * (v[0]).type(weight.dtype).to(weight.device)
+                    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)
                elif len(v) == 4: #lora/locon
                    mat1 = v[0]
                    mat2 = v[1]