diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-06 09:08:48 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-06 09:08:59 +0000 |
| commit | 5f24b7bcf4a074fbdec757617fcd1bc82e76551b (patch) | |
| tree | b4f1ccfb1718d667de448249882f385d4b9aafae /modules/sd_samplers.py | |
| parent | 6e7057b31b9762a9720282c7da486e4f264dee28 (diff) | |
| download | stable-diffusion-webui-gfx803-5f24b7bcf4a074fbdec757617fcd1bc82e76551b.tar.gz stable-diffusion-webui-gfx803-5f24b7bcf4a074fbdec757617fcd1bc82e76551b.tar.bz2 stable-diffusion-webui-gfx803-5f24b7bcf4a074fbdec757617fcd1bc82e76551b.zip | |
option to let users select which samplers they want to hide
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index d27c547b..2e1f7715 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -32,12 +32,27 @@ samplers_data_k_diffusion = [ if hasattr(k_diffusion.sampling, funcname)
]
-samplers = [
+all_samplers = [
*samplers_data_k_diffusion,
SamplerData('DDIM', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.ddim.DDIMSampler, model), []),
SamplerData('PLMS', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.plms.PLMSSampler, model), []),
]
-samplers_for_img2img = [x for x in samplers if x.name not in ['PLMS', 'DPM fast', 'DPM adaptive']]
+
+samplers = []
+samplers_for_img2img = []
+
+
+def set_samplers():
+ global samplers, samplers_for_img2img
+
+ hidden = set(opts.hide_samplers)
+ hidden_img2img = set(opts.hide_samplers + ['PLMS', 'DPM fast', 'DPM adaptive'])
+
+ samplers = [x for x in all_samplers if x.name not in hidden]
+ samplers_for_img2img = [x for x in all_samplers if x.name not in hidden_img2img]
+
+
+set_samplers()
sampler_extra_params = {
'sample_euler': ['s_churn', 's_tmin', 's_tmax', 's_noise'],
|