Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

handle invalid samplers

Browse Source
This commit is contained in:
Vladimir Mandic
2023-08-27 07:22:35 +00:00
parent 2e75ce2610
commit f45adac44a
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers.py
+5 -5
View File
@@ -62,11 +62,11 @@ def create_sampler(name, model):
def set_samplers():
global samplers, samplers_for_img2img # pylint: disable=global-statement
shown_img2img = set(shared.opts.show_samplers)
if len(shared.opts.show_samplers) == 0:
shown = {'PLMS', 'UniPC'}
else:
shown = set(shared.opts.show_samplers + ['PLMS'])
shown = shared.opts.show_samplers
if type(shown) is not list or len(shown) == 0:
shown = ['UniPC']
shown_img2img = set(shown)
shown = set(shown + ['PLMS'])
samplers = [x for x in all_samplers if x.name in shown]
samplers_for_img2img = [x for x in all_samplers if x.name in shown_img2img]
samplers_map.clear()