Add dpmpp_3 schedulers

This commit is contained in:
Sergey Borisov
2024-07-29 22:52:15 +03:00
parent 86a92bb6b5
commit 55e810efa3
5 changed files with 31 additions and 11 deletions

View File

@ -32,6 +32,8 @@ export const zSchedulerField = z.enum([
'ddpm',
'dpmpp_2s',
'dpmpp_2m',
'dpmpp_3s',
'dpmpp_3m',
'dpmpp_2m_sde',
'dpmpp_sde',
'heun',
@ -43,6 +45,8 @@ export const zSchedulerField = z.enum([
'deis_k',
'dpmpp_2s_k',
'dpmpp_2m_k',
'dpmpp_3s_k',
'dpmpp_3m_k',
'dpmpp_2m_sde_k',
'dpmpp_sde_k',
'heun_k',

View File

@ -23,6 +23,8 @@ export const zSchedulerField = z.enum([
'ddpm',
'dpmpp_2s',
'dpmpp_2m',
'dpmpp_3s',
'dpmpp_3m',
'dpmpp_2m_sde',
'dpmpp_sde',
'heun',
@ -34,6 +36,8 @@ export const zSchedulerField = z.enum([
'deis_k',
'dpmpp_2s_k',
'dpmpp_2m_k',
'dpmpp_3s_k',
'dpmpp_3m_k',
'dpmpp_2m_sde_k',
'dpmpp_sde_k',
'heun_k',

View File

@ -59,6 +59,8 @@ export const SCHEDULER_OPTIONS: ComboboxOption[] = [
{ value: 'dpmpp_sde', label: 'DPM++ SDE' },
{ value: 'dpmpp_2s', label: 'DPM++ 2S' },
{ value: 'dpmpp_2m', label: 'DPM++ 2M' },
{ value: 'dpmpp_3s', label: 'DPM++ 3S' },
{ value: 'dpmpp_3m', label: 'DPM++ 3M' },
{ value: 'dpmpp_2m_sde', label: 'DPM++ 2M SDE' },
{ value: 'heun', label: 'Heun' },
{ value: 'kdpm_2', label: 'KDPM 2' },
@ -70,6 +72,8 @@ export const SCHEDULER_OPTIONS: ComboboxOption[] = [
{ value: 'dpmpp_sde_k', label: 'DPM++ SDE Karras' },
{ value: 'dpmpp_2s_k', label: 'DPM++ 2S Karras' },
{ value: 'dpmpp_2m_k', label: 'DPM++ 2M Karras' },
{ value: 'dpmpp_3s_k', label: 'DPM++ 3S Karras' },
{ value: 'dpmpp_3m_k', label: 'DPM++ 3M Karras' },
{ value: 'dpmpp_2m_sde_k', label: 'DPM++ 2M SDE Karras' },
{ value: 'heun_k', label: 'Heun Karras' },
{ value: 'kdpm_2_k', label: 'KDPM 2 Karras' },