Skip to content

GitLab

"ppocr/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "2e67b9ae8e47f1f98a802ad668302114e578247c"
Commit 6ad0a6d7 authored by comfyanonymous's avatar comfyanonymous
Browse files

Don't patch weights when multiplier is zero.

parent af15add9
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
comfy/sd.py
View file @ 6ad0a6d7
...@@ -359,10 +359,11 @@ class ModelPatcher: ...@@ -359,10 +359,11 @@ class ModelPatcher:
if len(v) == 1: if len(v) == 1:
w1 = v[0] w1 = v[0]
if w1.shape != weight.shape: if alpha != 0.0:
print("WARNING SHAPE MISMATCH {} WEIGHT NOT MERGED {} != {}".format(key, w1.shape, weight.shape)) if w1.shape != weight.shape:
else: print("WARNING SHAPE MISMATCH {} WEIGHT NOT MERGED {} != {}".format(key, w1.shape, weight.shape))
weight += alpha * w1.type(weight.dtype).to(weight.device) 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]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment