1580158
/
sd-webui-supermerger
mirror of https://github.com/hako-mikan/sd-webui-supermerger.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix add difference

pull/464/head
hako-mikan 2025-01-21 08:41:06 +09:00
parent 6f5ac21757
commit eed0ec79a9
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/mergers/mergers.py
View File

@ -355,10 +355,7 @@ def smerge(weights_a,weights_b,model_a,model_b,model_c,base_alpha,base_beta,mode
if stopmerge: return "STOPPED", *NON4
if not ("weight" in key or "bias" in key): continue
if key in theta_2:
if uselerp:
theta_1[key] = torch.lerp(theta_1[key].to(torch.float32), -theta_2[key].to(torch.float32), 1.0).to(theta_1[key].dtype)
else:
theta_1[key] = (theta_1[key].to(torch.float32) -theta_2[key].to(torch.float32)).to(theta_1[key].dtype)
theta_1[key] = (theta_1[key].to(torch.float32) -theta_2[key].to(torch.float32)).to(theta_1[key].dtype)
else:
theta_1[key] = torch.zeros_like(theta_1[key].to(torch.float16))