diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2024-01-01 13:28:58 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2024-01-01 13:28:58 +0000 |
| commit | ac0ecf3b4b9d147743c04f0ff4ddc4cf4595e11d (patch) | |
| tree | e02a744d39bccc8a112431fc9e9cb0b1b3c8b7ff /modules/shared_options.py | |
| parent | 0743ee9b3eda8dd4ceea625d710031577201f4ad (diff) | |
| download | stable-diffusion-webui-gfx803-ac0ecf3b4b9d147743c04f0ff4ddc4cf4595e11d.tar.gz stable-diffusion-webui-gfx803-ac0ecf3b4b9d147743c04f0ff4ddc4cf4595e11d.tar.bz2 stable-diffusion-webui-gfx803-ac0ecf3b4b9d147743c04f0ff4ddc4cf4595e11d.zip | |
option to convert VAE to bfloat16 (implementation of #9295)
Diffstat (limited to 'modules/shared_options.py')
| -rw-r--r-- | modules/shared_options.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/shared_options.py b/modules/shared_options.py index ce06f022..e813546f 100644 --- a/modules/shared_options.py +++ b/modules/shared_options.py @@ -177,6 +177,7 @@ For img2img, VAE is used to process user's input image before the sampling, and "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, infotext='VAE').info("choose VAE model: Automatic = use one with same filename as checkpoint; None = use VAE from checkpoint"),
"sd_vae_overrides_per_model_preferences": OptionInfo(True, "Selected VAE overrides per-model preferences").info("you can set per-model VAE either by editing user metadata for checkpoints, or by making the VAE have same name as checkpoint"),
+ "auto_vae_precision_bfloat16": OptionInfo(False, "Automatically convert VAE to bfloat16").info("triggers when a tensor with NaNs is produced in VAE; disabling the option in this case will result in a black square image; if enabled, overrides the option below"),
"auto_vae_precision": OptionInfo(True, "Automatically revert VAE to 32-bit floats").info("triggers when a tensor with NaNs is produced in VAE; disabling the option in this case will result in a black square image"),
"sd_vae_encode_method": OptionInfo("Full", "VAE type for encode", gr.Radio, {"choices": ["Full", "TAESD"]}, infotext='VAE Encoder').info("method to encode image to latent (use in img2img, hires-fix or inpaint mask)"),
"sd_vae_decode_method": OptionInfo("Full", "VAE type for decode", gr.Radio, {"choices": ["Full", "TAESD"]}, infotext='VAE Decoder').info("method to decode latent to image"),
|