mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-02-20 06:43:57 +00:00
2fc1708a59
Pending: CFG++ Samplers, ODE Samplers The latter is probably easy to implement, the former needs modifications in sd_samplers_cfg_denoiser.py
27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
from modules import sd_samplers_kdiffusion, sd_samplers_common
|
|
from backend.modules import k_diffusion_extra
|
|
|
|
|
|
class AlterSampler(sd_samplers_kdiffusion.KDiffusionSampler):
|
|
def __init__(self, sd_model, sampler_name):
|
|
self.sampler_name = sampler_name
|
|
self.unet = sd_model.forge_objects.unet
|
|
sampler_function = getattr(k_diffusion_extra, "sample_{}".format(sampler_name))
|
|
super().__init__(sampler_function, sd_model, None)
|
|
|
|
|
|
def build_constructor(sampler_name):
|
|
def constructor(m):
|
|
return AlterSampler(m, sampler_name)
|
|
|
|
return constructor
|
|
|
|
|
|
samplers_data_alter = [
|
|
sd_samplers_common.SamplerData('DDPM', build_constructor(sampler_name='ddpm'), ['ddpm'], {}),
|
|
sd_samplers_common.SamplerData('HeunPP2', build_constructor(sampler_name='heunpp2'), ['heunpp2'], {}),
|
|
sd_samplers_common.SamplerData('IPNDM', build_constructor(sampler_name='ipndm'), ['ipndm'], {}),
|
|
sd_samplers_common.SamplerData('IPNDM_V', build_constructor(sampler_name='ipndm_v'), ['ipndm_v'], {}),
|
|
sd_samplers_common.SamplerData('DEIS', build_constructor(sampler_name='deis'), ['deis'], {}),
|
|
]
|