stable-diffusion-webui

1
from modules import sd_samplers_kdiffusion, sd_samplers_timesteps, sd_samplers_lcm, shared
2

3
# imports for functions that previously were here and are used by other modules
4
from modules.sd_samplers_common import samples_to_image_grid, sample_to_image  # noqa: F401
5

6
all_samplers = [
7
    *sd_samplers_kdiffusion.samplers_data_k_diffusion,
8
    *sd_samplers_timesteps.samplers_data_timesteps,
9
    *sd_samplers_lcm.samplers_data_lcm,
10
]
11
all_samplers_map = {x.name: x for x in all_samplers}
12

13
samplers = []
14
samplers_for_img2img = []
15
samplers_map = {}
16
samplers_hidden = {}
17

18

19
def find_sampler_config(name):
20
    if name is not None:
21
        config = all_samplers_map.get(name, None)
22
    else:
23
        config = all_samplers[0]
24

25
    return config
26

27

28
def create_sampler(name, model):
29
    config = find_sampler_config(name)
30

31
    assert config is not None, f'bad sampler name: {name}'
32

33
    if model.is_sdxl and config.options.get("no_sdxl", False):
34
        raise Exception(f"Sampler {config.name} is not supported for SDXL")
35

36
    sampler = config.constructor(model)
37
    sampler.config = config
38

39
    return sampler
40

41

42
def set_samplers():
43
    global samplers, samplers_for_img2img, samplers_hidden
44

45
    samplers_hidden = set(shared.opts.hide_samplers)
46
    samplers = all_samplers
47
    samplers_for_img2img = all_samplers
48

49
    samplers_map.clear()
50
    for sampler in all_samplers:
51
        samplers_map[sampler.name.lower()] = sampler.name
52
        for alias in sampler.aliases:
53
            samplers_map[alias.lower()] = sampler.name
54

55

56
def visible_sampler_names():
57
    return [x.name for x in samplers if x.name not in samplers_hidden]
58

59

60
set_samplers()
61