diff --git a/comfy/sd.py b/comfy/sd.py index 526dc531110..4bc9a15fa8a 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -359,10 +359,11 @@ def calculate_weight(self, patches, weight, key): 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]