mirror of
https://github.com/vladmandic/automatic
synced 2026-09-13 10:08:43 +02:00
Vladimir Mandic
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
from modules import sd_samplers_compvis, sd_samplers_kdiffusion, sd_samplers_diffusers, shared
|
|
from modules.sd_samplers_common import samples_to_image_grid, sample_to_image # pylint: disable=unused-import
|
|
|
|
|
|
all_samplers = []
|
|
all_samplers = []
|
|
all_samplers_map = {}
|
|
samplers = all_samplers
|
|
samplers_for_img2img = all_samplers
|
|
samplers_map = {}
|
|
|
|
def list_samplers(backend_name = shared.backend):
|
|
global all_samplers # pylint: disable=global-statement
|
|
global all_samplers_map # pylint: disable=global-statement
|
|
global samplers # pylint: disable=global-statement
|
|
global samplers_for_img2img # pylint: disable=global-statement
|
|
global samplers_map # pylint: disable=global-statement
|
|
if backend_name == shared.Backend.ORIGINAL:
|
|
all_samplers = [*sd_samplers_kdiffusion.samplers_data_k_diffusion, *sd_samplers_compvis.samplers_data_compvis]
|
|
else:
|
|
all_samplers = [*sd_samplers_diffusers.samplers_data_diffusers]
|
|
all_samplers_map = {x.name: x for x in all_samplers}
|
|
samplers = all_samplers
|
|
samplers_for_img2img = all_samplers
|
|
samplers_map = {}
|
|
shared.log.debug(f'Enumerated samplers: {len(all_samplers)}')
|
|
|
|
list_samplers()
|
|
|
|
|
|
def find_sampler_config(name):
|
|
if name is not None and name != '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)
|
|
if config is None:
|
|
shared.log.error(f'Attempting to use unknown sampler: {name}')
|
|
config = all_samplers[0]
|
|
if shared.backend == shared.Backend.ORIGINAL:
|
|
sampler = config.constructor(model)
|
|
sampler.config = config
|
|
return sampler
|
|
elif shared.backend == shared.Backend.DIFFUSERS:
|
|
sampler = config.constructor(model)
|
|
model.scheduler = sampler.sampler
|
|
return sampler.sampler
|
|
else:
|
|
return None
|
|
|
|
|
|
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'])
|
|
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()
|
|
for sampler in all_samplers:
|
|
samplers_map[sampler.name.lower()] = sampler.name
|
|
for alias in sampler.aliases:
|
|
samplers_map[alias.lower()] = sampler.name
|