diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index 9292192f..0e4072e2 100644 --- a/modules/sd_samplers_kdiffusion.py +++ b/modules/sd_samplers_kdiffusion.py @@ -27,6 +27,10 @@ samplers_k_diffusion = [ ('DPM fast', 'sample_dpm_fast', ['k_dpm_fast'], {"uses_ensd": True}), ('DPM adaptive', 'sample_dpm_adaptive', ['k_dpm_ad'], {"uses_ensd": True}), ('Restart', sd_samplers_extra.restart_sampler, ['restart'], {'scheduler': 'karras', "second_order": True}), + ('HeunPP2', 'sample_heunpp2', ['heunpp2'], {}), + ('IPNDM', 'sample_ipndm', ['ipndm'], {}), + ('IPNDM_V', 'sample_ipndm_v', ['ipndm_v'], {}), + ('DEIS', 'sample_deis', ['deis'], {}), ] diff --git a/modules_forge/alter_samplers.py b/modules_forge/alter_samplers.py index af216d34..15cbe2f0 100644 --- a/modules_forge/alter_samplers.py +++ b/modules_forge/alter_samplers.py @@ -19,8 +19,4 @@ def build_constructor(sampler_name): 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'], {}), ]