diff --git a/src/config.ts b/src/config.ts index 5f87954..e60ab55 100644 --- a/src/config.ts +++ b/src/config.ts @@ -56,28 +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_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', - '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 = {