From 8c7db614ba22641d9f95cccf744293c97ef3f189 Mon Sep 17 00:00:00 2001 From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com> Date: Tue, 20 Aug 2024 15:00:56 +0100 Subject: [PATCH 1/2] Update alter_samplers.py move new samplers from here --- modules_forge/alter_samplers.py | 4 ---- 1 file changed, 4 deletions(-) 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'], {}), ] From c127e60cf0d98ac56d18eadf418fa9b1b4b12be8 Mon Sep 17 00:00:00 2001 From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com> Date: Tue, 20 Aug 2024 15:01:58 +0100 Subject: [PATCH 2/2] Update sd_samplers_kdiffusion.py add new samplers here --- modules/sd_samplers_kdiffusion.py | 4 ++++ 1 file changed, 4 insertions(+) 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'], {}), ]