aboutsummaryrefslogtreecommitdiffstats
path: root/modules/shared.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-10-15 07:35:26 +0000
committerGitHub <noreply@github.com>2022-10-15 07:35:26 +0000
commitd13ce89e203d76ab2b54a3406a93a5e4304f529e (patch)
treec3ef9ef64f9959f2820a8083420f4f9b7486d636 /modules/shared.py
parent6a4e84671016d38c10a55fedcdf09321dba737ae (diff)
parentaf144ebdc70ef6f006bebd47af7a8d3180665538 (diff)
downloadstable-diffusion-webui-gfx803-d13ce89e203d76ab2b54a3406a93a5e4304f529e.tar.gz
stable-diffusion-webui-gfx803-d13ce89e203d76ab2b54a3406a93a5e4304f529e.tar.bz2
stable-diffusion-webui-gfx803-d13ce89e203d76ab2b54a3406a93a5e4304f529e.zip
Merge pull request #2573 from raefu/ckpt-cache
add --ckpt-cache option for faster model switching
Diffstat (limited to 'modules/shared.py')
-rw-r--r--modules/shared.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py
index d41a7ab3..aa69bedf 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -242,6 +242,7 @@ options_templates.update(options_section(('training', "Training"), {
options_templates.update(options_section(('sd', "Stable Diffusion"), {
"sd_model_checkpoint": OptionInfo(None, "Stable Diffusion checkpoint", gr.Dropdown, lambda: {"choices": modules.sd_models.checkpoint_tiles()}, refresh=sd_models.list_models),
+ "sd_checkpoint_cache": OptionInfo(0, "Checkpoints to cache in RAM", gr.Slider, {"minimum": 0, "maximum": 10, "step": 1}),
"sd_hypernetwork": OptionInfo("None", "Hypernetwork", gr.Dropdown, lambda: {"choices": ["None"] + [x for x in hypernetworks.keys()]}, refresh=reload_hypernetworks),
"sd_hypernetwork_strength": OptionInfo(1.0, "Hypernetwork strength", gr.Slider, {"minimum": 0.0, "maximum": 1.0, "step": 0.001}),
"img2img_color_correction": OptionInfo(False, "Apply color correction to img2img results to match original colors."),