feat: add new stable diffusion samplers

koishijs · Feb 14, 2024 · b3b329c · b3b329c
1 parent dcfda09
commit b3b329c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/config.ts b/src/config.ts
@@ -66,6 +66,10 @@ export namespace sampler {
     'k_dpmpp_2m': 'DPM++ 2M',
     'k_dpmpp_sde': 'DPM++ SDE',
     'k_dpmpp_2m_sde': 'DPM++ 2M SDE',
+    'k_dpmpp_2m_sde_heun': 'DPM++ 2M SDE Heun',
+    'k_dpmpp_2m_sde_heun_exp': 'DPM++ 2M SDE Heun Exponential',
+    'k_dpmpp_3m_sde': 'DPM++ 3M SDE',
+    'k_dpmpp_3m_sde_exp': 'DPM++ 3M SDE Exponential',
     'k_dpm_fast': 'DPM fast',
     'k_dpm_ad': 'DPM adaptive',
     'k_lms_ka': 'LMS Karras',
@@ -75,6 +79,8 @@ export namespace sampler {
     'k_dpmpp_2m_ka': 'DPM++ 2M Karras',
     'k_dpmpp_sde_ka': 'DPM++ SDE Karras',
     'k_dpmpp_2m_sde_ka': 'DPM++ 2M SDE Karras',
+    'k_dpmpp_2m_sde_heun_ka': 'DPM++ 2M SDE Heun Karras',
+    'k_dpmpp_3m_sde_ka': 'DPM++ 3M SDE Karras',
     'ddim': 'DDIM',
     'plms': 'PLMS',
     'unipc': 'UniPC',