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

Merge pull request #45 from space-nuko/save-recipe-metadata 

Add no metadata option
ver7
hako-mikan 2023-04-06 19:36:31 +09:00 committed by GitHub
parent 3bff3abe08 20b587f296
commit 2794ce4dae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/mergers/mergers.py
View File

@ -97,7 +97,7 @@ def smerge(weights_a,weights_b,model_a,model_b,model_c,base_alpha,base_beta,mode
# mode select booleans
save = True if sevemodes[0] in save_sets else False
usebeta = modes[2] in mode or modes[3] in mode
save_metadata = "safetensors" in save_sets
save_metadata = "save metadata" in save_sets
metadata = {"format": "pt", "sd_merge_models": {}, "sd_merge_recipe": None}
if not useblocks:

2
scripts/supermerger.py
View File

@ -80,7 +80,7 @@ def on_ui_tabs():
gen = gr.Button(elem_id="model_merger_merge", value="Gen",variant='primary')
with gr.Row():
with gr.Column(scale = 4):
save_sets = gr.CheckboxGroup(["save model", "overwrite","safetensors","fp16"], value=["safetensors"], label="save settings")
save_sets = gr.CheckboxGroup(["save model", "overwrite","safetensors","fp16","save metadata"], value=["safetensors"], label="save settings")
with gr.Column(scale = 2):
id_sets = gr.CheckboxGroup(["image", "PNG info"], label="write merged model ID to")
with gr.Row():