diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-04-29 14:15:01 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-29 14:15:01 +0000 |
| commit | 8987764395f72d85776f91512391ccd4688e47bb (patch) | |
| tree | 312f61445d8cbffb4de7949dcb0f82fbeb659658 /modules/ui.py | |
| parent | 31dbec6b761d912ad986727068c2f3cc4d39a23d (diff) | |
| parent | fbaf6e4fd897fa1f3e3f747f1d699c240cad76a0 (diff) | |
| download | stable-diffusion-webui-gfx803-8987764395f72d85776f91512391ccd4688e47bb.tar.gz stable-diffusion-webui-gfx803-8987764395f72d85776f91512391ccd4688e47bb.tar.bz2 stable-diffusion-webui-gfx803-8987764395f72d85776f91512391ccd4688e47bb.zip | |
Merge pull request #9312 from space-nuko/save-merge-recipe
Embed model merge metadata in .safetensors file
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/ui.py b/modules/ui.py index dab94b93..74a7d315 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1019,8 +1019,9 @@ def create_ui(): interp_method.change(fn=update_interp_description, inputs=[interp_method], outputs=[interp_description])
with FormRow():
- checkpoint_format = gr.Radio(choices=["ckpt", "safetensors"], value="ckpt", label="Checkpoint format", elem_id="modelmerger_checkpoint_format")
+ checkpoint_format = gr.Radio(choices=["ckpt", "safetensors"], value="safetensors", label="Checkpoint format", elem_id="modelmerger_checkpoint_format")
save_as_half = gr.Checkbox(value=False, label="Save as float16", elem_id="modelmerger_save_as_half")
+ save_metadata = gr.Checkbox(value=True, label="Save metadata (.safetensors only)", elem_id="modelmerger_save_metadata")
with FormRow():
with gr.Column():
@@ -1658,6 +1659,7 @@ def create_ui(): config_source,
bake_in_vae,
discard_weights,
+ save_metadata,
],
outputs=[
primary_model_name,
|