diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-16 08:54:02 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-16 08:54:02 +0000 |
| commit | cdac5ace1456ba779d5a0171ff8757f31955bfee (patch) | |
| tree | e9c13ec3186e182209c7258a8c7305ef2ac68db1 /modules/sd_samplers.py | |
| parent | 3d76eabbca3adb711787d1802d6b61c0971b4bc0 (diff) | |
| download | stable-diffusion-webui-gfx803-cdac5ace1456ba779d5a0171ff8757f31955bfee.tar.gz stable-diffusion-webui-gfx803-cdac5ace1456ba779d5a0171ff8757f31955bfee.tar.bz2 stable-diffusion-webui-gfx803-cdac5ace1456ba779d5a0171ff8757f31955bfee.zip | |
suppress ENSD infotext for samplers that don't use it
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 4f1bf21d..f22aad8f 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -14,12 +14,18 @@ samplers_for_img2img = [] samplers_map = {}
-def create_sampler(name, model):
+def find_sampler_config(name):
if name is not None:
config = all_samplers_map.get(name, None)
else:
config = all_samplers[0]
+ return config
+
+
+def create_sampler(name, model):
+ config = find_sampler_config(name)
+
assert config is not None, f'bad sampler name: {name}'
sampler = config.constructor(model)
|