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

Update mergers.py

pull/103/head
hako-mikan 2023-05-23 23:24:34 +09:00 committed by GitHub
parent 1b40c093de
commit c56fc54c3d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/mergers/mergers.py
View File

@ -114,7 +114,7 @@ def smerge(weights_a,weights_b,model_a,model_b,model_c,base_alpha,base_beta,mode
# mode select booleans
save = True if SAVEMODES[0] in save_sets else False
usebeta = MODES[2] in mode or MODES[3] in mode
usebeta = MODES[2] in mode or MODES[3] in mode or calcmode == "tensor"
save_metadata = "save metadata" in save_sets
metadata = {"format": "pt"}
@ -128,6 +128,7 @@ def smerge(weights_a,weights_b,model_a,model_b,model_c,base_alpha,base_beta,mode
model_a = namefromhash(model_a)
model_b = namefromhash(model_b)
model_c = namefromhash(model_c)
theta_2 = {}
caster(mergedmodel,False)
@ -242,7 +243,7 @@ def smerge(weights_a,weights_b,model_a,model_b,model_c,base_alpha,base_beta,mode
for key in (tqdm(theta_0.keys(), desc="Stage 1/2") if not False else theta_0.keys()):
if stopmerge: return "STOPPED", *non4
if "model" in key and key in theta_1:
if usebeta and not key in theta_2:
if usebeta and (not key in theta_2) and (not theta_2 == {}) :
continue
weight_index = -1