diff --git a/scripts/mergers/mergers.py b/scripts/mergers/mergers.py index fdc2542..a146a59 100644 --- a/scripts/mergers/mergers.py +++ b/scripts/mergers/mergers.py @@ -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: diff --git a/scripts/supermerger.py b/scripts/supermerger.py index 96b4aef..5287069 100644 --- a/scripts/supermerger.py +++ b/scripts/supermerger.py @@ -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():