1580158
/
sdweb-merge-block-weighted-gui
mirror of https://github.com/bbc-mc/sdweb-merge-block-weighted-gui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: correctly save log of merged output checkpoint hash model_O_hash

fix_update
bbc_mc 2023-01-08 13:25:00 +09:00
parent e18a67a95d
commit 6bbfdb35d0
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/mbw/ui_mbw.py
View File

@ -1,5 +1,6 @@
import gradio as gr
import os
import re
from modules import sd_models, shared
from tqdm import tqdm
@ -145,8 +146,8 @@ def on_ui_tabs():
# save log to history.tsv
sd_models.list_models()
model_O_info = sd_models.get_closet_checkpoint_match(model_O)
model_O_hash = "" if not model_O_info else model_O_info.hash
model_O_info = sd_models.get_closet_checkpoint_match(os.path.basename(_output))
model_O_hash = "" if model_O_info is None else model_O_info.hash
_names = presetWeights.find_names_by_weight(_weights)
if _names and len(_names) > 0:
weight_name = _names[0]

3
scripts/mbw_each/ui_mbw_each.py
View File

@ -365,7 +365,8 @@ def on_ui_tabs():
# save log to history.tsv
model_O_info = sd_models.get_closet_checkpoint_match(model_O)
sd_models.list_models()
model_O_info = sd_models.get_closet_checkpoint_match(os.path.basename(output_file))
model_O_hash = "" if not model_O_info else model_O_info.hash
_names = presetWeights.find_names_by_weight(weight_A)
if _names and len(_names) > 0: