Merge pull request #12521 from catboxanon/feat/more-s-noise

Add `s_noise` param to more samplers

1 files changed, 9 insertions, 0 deletions

diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index 9a89e7fa..0bacfe8d 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -22,6 +22,9 @@ samplers_k_diffusion = [
     ('DPM++ 2M', 'sample_dpmpp_2m', ['k_dpmpp_2m'], {}),

     ('DPM++ SDE', 'sample_dpmpp_sde', ['k_dpmpp_sde'], {"second_order": True, "brownian_noise": True}),

     ('DPM++ 2M SDE', 'sample_dpmpp_2m_sde', ['k_dpmpp_2m_sde_ka'], {"brownian_noise": True}),

+    ('DPM++ 3M SDE', 'sample_dpmpp_3m_sde', ['k_dpmpp_3m_sde'], {'discard_next_to_last_sigma': True, "brownian_noise": True}),

+    ('DPM++ 3M SDE Karras', 'sample_dpmpp_3m_sde', ['k_dpmpp_3m_sde_ka'], {'scheduler': 'karras', 'discard_next_to_last_sigma': True, "brownian_noise": True}),

+    ('DPM++ 3M SDE Exponential', 'sample_dpmpp_3m_sde', ['k_dpmpp_3m_sde_exp'], {'scheduler': 'exponential', 'discard_next_to_last_sigma': True, "brownian_noise": True}),

     ('DPM fast', 'sample_dpm_fast', ['k_dpm_fast'], {"uses_ensd": True}),

     ('DPM adaptive', 'sample_dpm_adaptive', ['k_dpm_ad'], {"uses_ensd": True}),

     ('LMS Karras', 'sample_lms', ['k_lms_ka'], {'scheduler': 'karras'}),

@@ -42,6 +45,12 @@ sampler_extra_params = {
     'sample_euler': ['s_churn', 's_tmin', 's_tmax', 's_noise'],

     'sample_heun': ['s_churn', 's_tmin', 's_tmax', 's_noise'],

     'sample_dpm_2': ['s_churn', 's_tmin', 's_tmax', 's_noise'],

+    'sample_dpm_fast': ['s_noise'],

+    'sample_dpm_2_ancestral': ['s_noise'],

+    'sample_dpmpp_2s_ancestral': ['s_noise'],

+    'sample_dpmpp_sde': ['s_noise'],

+    'sample_dpmpp_2m_sde': ['s_noise'],

+    'sample_dpmpp_3m_sde': ['s_noise'],

 }

 

 k_diffusion_samplers_map = {x.name: x for x in samplers_data_k_diffusion}