feat: add new stable diffusion samplers (#230)

This commit is contained in:
Maiko Sinkyaet Tan 2024-02-17 00:29:46 +08:00 committed by GitHub
parent 6e654503a3
commit 9e5b75c115
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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 = {