diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-27 12:47:33 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-27 12:47:33 +0000 |
| commit | 339b5315700a469f4a9f0d5afc08ca2aca60c579 (patch) | |
| tree | 154d6ed405883ca47e23d67e6d3f7699ff4e23e3 /modules/shared.py | |
| parent | a6e653be26cc05f4438145fa0082816e9fbbf5fc (diff) | |
| download | stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.tar.gz stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.tar.bz2 stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.zip | |
custom unet support
Diffstat (limited to 'modules/shared.py')
| -rw-r--r-- | modules/shared.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py index 0897f937..a5e7824a 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -403,6 +403,7 @@ options_templates.update(options_section(('sd', "Stable Diffusion"), { "sd_vae_checkpoint_cache": OptionInfo(0, "VAE Checkpoints to cache in RAM", gr.Slider, {"minimum": 0, "maximum": 10, "step": 1}),
"sd_vae": OptionInfo("Automatic", "SD VAE", gr.Dropdown, lambda: {"choices": shared_items.sd_vae_items()}, refresh=shared_items.refresh_vae_list).info("choose VAE model: Automatic = use one with same filename as checkpoint; None = use VAE from checkpoint"),
"sd_vae_as_default": OptionInfo(True, "Ignore selected VAE for stable diffusion checkpoints that have their own .vae.pt next to them"),
+ "sd_unet": OptionInfo("Automatic", "SD Unet", gr.Dropdown, lambda: {"choices": shared_items.sd_unet_items()}, refresh=shared_items.refresh_unet_list).info("choose Unet model: Automatic = use one with same filename as checkpoint; None = use Unet from checkpoint"),
"inpainting_mask_weight": OptionInfo(1.0, "Inpainting conditioning mask strength", gr.Slider, {"minimum": 0.0, "maximum": 1.0, "step": 0.01}),
"initial_noise_multiplier": OptionInfo(1.0, "Noise multiplier for img2img", gr.Slider, {"minimum": 0.5, "maximum": 1.5, "step": 0.01}),
"img2img_color_correction": OptionInfo(False, "Apply color correction to img2img results to match original colors."),
|