suppress ENSD infotext for samplers that don't use it

This commit is contained in:
AUTOMATIC
2023-05-16 11:54:02 +03:00
parent 3d76eabbca
commit cdac5ace14
5 changed files with 48 additions and 16 deletions

View File

@@ -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)