lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-13 08:59:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

add sampler

Browse Source
This commit is contained in:
layerdiffusion
2024-08-03 15:46:27 -07:00
parent b8b29a2372
commit 3ecdcee5a8
2 changed files with 58 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
modules_forge/forge_alter_samplers.py
View File

@@ -1,23 +1,22 @@
# from modules import sd_samplers_kdiffusion, sd_samplers_common
# from ldm_patched.k_diffusion import sampling as k_diffusion_sampling
#
#
# 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_sampling, "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
#
#
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('DDPM', build_constructor(sampler_name='ddpm'), ['ddpm'], {}),
]