aboutsummaryrefslogtreecommitdiffstats
path: root/modules/shared.py
diff options
context:
space:
mode:
authorspace-nuko <24979496+space-nuko@users.noreply.github.com>2023-02-10 13:27:05 +0000
committerspace-nuko <24979496+space-nuko@users.noreply.github.com>2023-02-10 13:27:05 +0000
commit79ffb9453f8eddbdd4e316b9d9c75812b0eea4e1 (patch)
tree03793e161ca395aa60b44c046210a8b88a8bb73a /modules/shared.py
parentc88dcc20d495dab4be2692bdff30277112dbe416 (diff)
downloadstable-diffusion-webui-gfx803-79ffb9453f8eddbdd4e316b9d9c75812b0eea4e1.tar.gz
stable-diffusion-webui-gfx803-79ffb9453f8eddbdd4e316b9d9c75812b0eea4e1.tar.bz2
stable-diffusion-webui-gfx803-79ffb9453f8eddbdd4e316b9d9c75812b0eea4e1.zip
Add UniPC sampler settings
Diffstat (limited to 'modules/shared.py')
-rw-r--r--modules/shared.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py
index 79fbf724..34242073 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -480,6 +480,11 @@ options_templates.update(options_section(('sampler-params', "Sampler parameters"
's_noise': OptionInfo(1.0, "sigma noise", gr.Slider, {"minimum": 0.0, "maximum": 1.0, "step": 0.01}),
'eta_noise_seed_delta': OptionInfo(0, "Eta noise seed delta", gr.Number, {"precision": 0}),
'always_discard_next_to_last_sigma': OptionInfo(False, "Always discard next-to-last sigma"),
+ 'uni_pc_variant': OptionInfo("bh1", "UniPC variant", gr.Radio, {"choices": ["bh1", "vary_coeff"]}),
+ 'uni_pc_skip_type': OptionInfo("time_uniform", "UniPC skip type", gr.Radio, {"choices": ["time_uniform", "time_quadratic", "logSNR"]}),
+ 'uni_pc_order': OptionInfo(3, "UniPC order (must be < sampling steps)", gr.Slider, {"minimum": 1, "maximum": 150 - 1, "step": 1}),
+ 'uni_pc_thresholding': OptionInfo(False, "UniPC thresholding"),
+ 'uni_pc_lower_order_final': OptionInfo(True, "UniPC lower order final"),
}))
options_templates.update(options_section(('postprocessing', "Postprocessing"), {