diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-11-27 12:55:03 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-27 12:55:03 +0300 |
| commit | d860b56c219a45b274c2147521b2a5823ee90a15 (patch) | |
| tree | 7cd692e17d8ff273916c52d30c77370f1871de54 /modules | |
| parent | b48b7999c86fd6d7f006f76adf5a484175782c37 (diff) | |
| parent | 6df49457183fedec635d30dd0c611844b0932c85 (diff) | |
Merge pull request #4961 from uservar/DPM++SDE
Add DPM++ SDE sampler
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/sd_samplers.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 4edd8c60..43ce34eb 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -26,6 +26,7 @@ samplers_k_diffusion = [ ('DPM2 a', 'sample_dpm_2_ancestral', ['k_dpm_2_a'], {}),
('DPM++ 2S a', 'sample_dpmpp_2s_ancestral', ['k_dpmpp_2s_a'], {}),
('DPM++ 2M', 'sample_dpmpp_2m', ['k_dpmpp_2m'], {}),
+ ('DPM++ SDE', 'sample_dpmpp_sde', ['k_dpmpp_sde'], {}),
('DPM fast', 'sample_dpm_fast', ['k_dpm_fast'], {}),
('DPM adaptive', 'sample_dpm_adaptive', ['k_dpm_ad'], {}),
('LMS Karras', 'sample_lms', ['k_lms_ka'], {'scheduler': 'karras'}),
@@ -33,6 +34,7 @@ samplers_k_diffusion = [ ('DPM2 a Karras', 'sample_dpm_2_ancestral', ['k_dpm_2_a_ka'], {'scheduler': 'karras'}),
('DPM++ 2S a Karras', 'sample_dpmpp_2s_ancestral', ['k_dpmpp_2s_a_ka'], {'scheduler': 'karras'}),
('DPM++ 2M Karras', 'sample_dpmpp_2m', ['k_dpmpp_2m_ka'], {'scheduler': 'karras'}),
+ ('DPM++ SDE Karras', 'sample_dpmpp_sde', ['k_dpmpp_sde_ka'], {'scheduler': 'karras'}),
]
samplers_data_k_diffusion = [
|