mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
73 lines
1.7 KiB
TypeScript
73 lines
1.7 KiB
TypeScript
import { SchedulerParam } from 'features/parameters/store/parameterZodSchemas';
|
|
|
|
// zod needs the array to be `as const` to infer the type correctly
|
|
// this is the source of the `SchedulerParam` type, which is generated by zod
|
|
export const SCHEDULER_NAMES_AS_CONST = [
|
|
'euler',
|
|
'deis',
|
|
'ddim',
|
|
'ddpm',
|
|
'dpmpp_2s',
|
|
'dpmpp_2m',
|
|
'dpmpp_2m_sde',
|
|
'dpmpp_sde',
|
|
'heun',
|
|
'kdpm_2',
|
|
'lms',
|
|
'pndm',
|
|
'unipc',
|
|
'euler_k',
|
|
'dpmpp_2s_k',
|
|
'dpmpp_2m_k',
|
|
'dpmpp_2m_sde_k',
|
|
'dpmpp_sde_k',
|
|
'heun_k',
|
|
'lms_k',
|
|
'euler_a',
|
|
'kdpm_2_a',
|
|
] as const;
|
|
|
|
export const DEFAULT_SCHEDULER_NAME = 'euler';
|
|
|
|
export const SCHEDULER_NAMES: SchedulerParam[] = [...SCHEDULER_NAMES_AS_CONST];
|
|
|
|
export const SCHEDULER_LABEL_MAP: Record<SchedulerParam, string> = {
|
|
euler: 'Euler',
|
|
deis: 'DEIS',
|
|
ddim: 'DDIM',
|
|
ddpm: 'DDPM',
|
|
dpmpp_sde: 'DPM++ SDE',
|
|
dpmpp_2s: 'DPM++ 2S',
|
|
dpmpp_2m: 'DPM++ 2M',
|
|
dpmpp_2m_sde: 'DPM++ 2M SDE',
|
|
heun: 'Heun',
|
|
kdpm_2: 'KDPM 2',
|
|
lms: 'LMS',
|
|
pndm: 'PNDM',
|
|
unipc: 'UniPC',
|
|
euler_k: 'Euler Karras',
|
|
dpmpp_sde_k: 'DPM++ SDE Karras',
|
|
dpmpp_2s_k: 'DPM++ 2S Karras',
|
|
dpmpp_2m_k: 'DPM++ 2M Karras',
|
|
dpmpp_2m_sde_k: 'DPM++ 2M SDE Karras',
|
|
heun_k: 'Heun Karras',
|
|
lms_k: 'LMS Karras',
|
|
euler_a: 'Euler Ancestral',
|
|
kdpm_2_a: 'KDPM 2 Ancestral',
|
|
};
|
|
|
|
export type Scheduler = (typeof SCHEDULER_NAMES)[number];
|
|
|
|
// Valid upscaling levels
|
|
export const UPSCALING_LEVELS: Array<{ label: string; value: string }> = [
|
|
{ label: '2x', value: '2' },
|
|
{ label: '4x', value: '4' },
|
|
];
|
|
export const NUMPY_RAND_MIN = 0;
|
|
|
|
export const NUMPY_RAND_MAX = 2147483647;
|
|
|
|
export const FACETOOL_TYPES = ['gfpgan', 'codeformer'] as const;
|
|
|
|
export const NODE_MIN_WIDTH = 250;
|