mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-06-07 21:20:49 +00:00
68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
from modules import sd_samplers_kdiffusion, sd_samplers_timesteps, sd_samplers_lcm, shared, sd_samplers_common
|
|
|
|
# imports for functions that previously were here and are used by other modules
|
|
samples_to_image_grid = sd_samplers_common.samples_to_image_grid
|
|
sample_to_image = sd_samplers_common.sample_to_image
|
|
|
|
all_samplers = [
|
|
*sd_samplers_kdiffusion.samplers_data_k_diffusion,
|
|
*sd_samplers_timesteps.samplers_data_timesteps,
|
|
*sd_samplers_lcm.samplers_data_lcm,
|
|
]
|
|
all_samplers_map = {x.name: x for x in all_samplers}
|
|
|
|
samplers: list[sd_samplers_common.SamplerData] = []
|
|
samplers_for_img2img: list[sd_samplers_common.SamplerData] = []
|
|
samplers_map = {}
|
|
samplers_hidden = {}
|
|
|
|
|
|
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}'
|
|
|
|
if model.is_sdxl and config.options.get("no_sdxl", False):
|
|
raise Exception(f"Sampler {config.name} is not supported for SDXL")
|
|
|
|
sampler = config.constructor(model)
|
|
sampler.config = config
|
|
|
|
return sampler
|
|
|
|
|
|
def set_samplers():
|
|
global samplers, samplers_for_img2img, samplers_hidden
|
|
|
|
samplers_hidden = set(shared.opts.hide_samplers)
|
|
samplers = all_samplers
|
|
samplers_for_img2img = all_samplers
|
|
|
|
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
|
|
|
|
|
|
def visible_sampler_names():
|
|
return [x.name for x in samplers if x.name not in samplers_hidden]
|
|
|
|
|
|
def visible_samplers():
|
|
return [x for x in samplers if x.name not in samplers_hidden]
|
|
|
|
|
|
set_samplers()
|