diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-11-27 12:51:29 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-11-27 12:51:29 +0000 |
| commit | dac9b6f15de5e675053d9490a20e0457dcd1a23e (patch) | |
| tree | bab88aafd9834054efeddcf6eba7673dc4eba3ec /modules/ui.py | |
| parent | 6074175faa751dde933aa8e15cd687ca4e4b4a23 (diff) | |
| download | stable-diffusion-webui-gfx803-dac9b6f15de5e675053d9490a20e0457dcd1a23e.tar.gz stable-diffusion-webui-gfx803-dac9b6f15de5e675053d9490a20e0457dcd1a23e.tar.bz2 stable-diffusion-webui-gfx803-dac9b6f15de5e675053d9490a20e0457dcd1a23e.zip | |
add safetensors support for model merging #4869
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/modules/ui.py b/modules/ui.py index de2b5544..aa13978d 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1164,7 +1164,11 @@ def create_ui(wrap_gradio_gpu_call): custom_name = gr.Textbox(label="Custom Name (Optional)")
interp_amount = gr.Slider(minimum=0.0, maximum=1.0, step=0.05, label='Multiplier (M) - set to 0 to get model A', value=0.3)
interp_method = gr.Radio(choices=["Weighted sum", "Add difference"], value="Weighted sum", label="Interpolation Method")
- save_as_half = gr.Checkbox(value=False, label="Save as float16")
+
+ with gr.Row():
+ checkpoint_format = gr.Radio(choices=["ckpt", "safetensors"], value="ckpt", label="Checkpoint format")
+ save_as_half = gr.Checkbox(value=False, label="Save as float16")
+
modelmerger_merge = gr.Button(elem_id="modelmerger_merge", label="Merge", variant='primary')
with gr.Column(variant='panel'):
@@ -1692,6 +1696,7 @@ def create_ui(wrap_gradio_gpu_call): interp_amount,
save_as_half,
custom_name,
+ checkpoint_format,
],
outputs=[
submit_result,
|