fix: some samplers not support karras

koishijs · Feb 15, 2024 · ea8057e · ea8057e
1 parent 785c664
commit ea8057e
Showing 1 changed file with 29 additions and 30 deletions.
diff --git a/src/config.ts b/src/config.ts
@@ -56,36 +56,35 @@ export namespace sampler {
   // https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/modules/sd_samplers_compvis.py#L12
   // https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/modules/sd_samplers_kdiffusion.py#L12
   export const sd = {
-    'k_euler_a': 'Euler a',
-    'k_euler': 'Euler',
-    'k_lms': 'LMS',
-    'k_heun': 'Heun',
-    'k_dpm_2': 'DPM2',
-    'k_dpm_2_a': 'DPM2 a',
-    'k_dpmpp_2s_a': 'DPM++ 2S a',
-    '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',
-    'k_dpm_2_ka': 'DPM2 Karras',
-    'k_dpm_2_a_ka': 'DPM2 a Karras',
-    'k_dpmpp_2s_a_ka': 'DPM++ 2S a Karras',
-    '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',
-    'restart': 'Restart',
-    'restart_ka': 'Restart Karras',
-    'ddim': 'DDIM',
-    'plms': 'PLMS',
-    'unipc': 'UniPC',
+    k_dpmpp_2m_ka: 'DPM++ 2M Karras',
+    k_dpmpp_sde_ka: 'DPM++ SDE Karras',
+    k_dpmpp_2m_sde_exp: 'DPM++ 2M SDE Exponential',
+    k_dpmpp_2m_sde_ka: 'DPM++ 2M SDE',
+    k_euler_a: 'Euler a',
+    k_euler: 'Euler',
+    k_lms: 'LMS',
+    k_heun: 'Heun',
+    k_dpm_2: 'DPM2',
+    k_dpm_2_a: 'DPM2 a',
+    k_dpmpp_2s_a: 'DPM++ 2S a',
+    k_dpmpp_2m: 'DPM++ 2M',
+    k_dpmpp_sde: 'DPM++ SDE',
+    k_dpmpp_2m_sde_heun: 'DPM++ 2M SDE Heun',
+    k_dpmpp_2m_sde_heun_ka: 'DPM++ 2M SDE Heun Karras',
+    k_dpmpp_2m_sde_heun_exp: 'DPM++ 2M SDE Heun Exponential',
+    k_dpmpp_3m_sde: 'DPM++ 3M SDE',
+    k_dpmpp_3m_sde_ka: 'DPM++ 3M SDE Karras',
+    k_dpmpp_3m_sde_exp: 'DPM++ 3M SDE Exponential',
+    k_dpm_fast: 'DPM fast',
+    k_dpm_ad: 'DPM adaptive',
+    k_lms_ka: 'LMS Karras',
+    k_dpm_2_ka: 'DPM2 Karras',
+    k_dpm_2_a_ka: 'DPM2 a Karras',
+    k_dpmpp_2s_a_ka: 'DPM++ 2S a Karras',
+    restart: 'Restart',
+    ddim: 'DDIM',
+    plms: 'PLMS',
+    unipc: 'UniPC',
   }
 
   export const horde = {