Browse Source

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

pull/793/head
comfyanonymous 1 year ago
parent
commit
9f83b098c9
  1. 6
      comfy/sd.py

6
comfy/sd.py

@ -390,7 +390,11 @@ class ModelPatcher:
weight *= strength_model weight *= strength_model
if len(v) == 1: 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 elif len(v) == 4: #lora/locon
mat1 = v[0] mat1 = v[0]
mat2 = v[1] mat2 = v[1]

Loading…
Cancel
Save