Commit 5dee0fa1 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

add a message about unsupported samplers

parent ac2d47ff

modules/sd_samplers.py

+3 −0
Original line number Diff line number Diff line
@@ -28,6 +28,9 @@ def create_sampler(name, model): 


    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

modules/sd_samplers_compvis.py

+3 −3
Original line number Diff line number Diff line
@@ -11,9 +11,9 @@ import modules.models.diffusion.uni_pc 




samplers_data_compvis = [

    sd_samplers_common.SamplerData('DDIM', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.ddim.DDIMSampler, model), [], {"default_eta_is_0": True, "uses_ensd": True}),

    sd_samplers_common.SamplerData('PLMS', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.plms.PLMSSampler, model), [], {}),

    sd_samplers_common.SamplerData('UniPC', lambda model: VanillaStableDiffusionSampler(modules.models.diffusion.uni_pc.UniPCSampler, model), [], {}),

    sd_samplers_common.SamplerData('DDIM', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.ddim.DDIMSampler, model), [], {"default_eta_is_0": True, "uses_ensd": True, "no_sdxl": True}),

    sd_samplers_common.SamplerData('PLMS', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.plms.PLMSSampler, model), [], {"no_sdxl": True}),

    sd_samplers_common.SamplerData('UniPC', lambda model: VanillaStableDiffusionSampler(modules.models.diffusion.uni_pc.UniPCSampler, model), [], {"no_sdxl": True}),

]

CRA Git | Maintained and supported by SUSTech CRA and CCSE