mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
62 lines
1.9 KiB
TypeScript
62 lines
1.9 KiB
TypeScript
import { SelectItem } from '@mantine/core';
|
|
|
|
export const SCHEDULERS: SelectItem[] = [
|
|
{ label: 'Euler', value: 'euler', group: 'Standard' },
|
|
{ label: 'DEIS', value: 'deis', group: 'Standard' },
|
|
{ label: 'DDIM', value: 'ddim', group: 'Standard' },
|
|
{ label: 'DDPM', value: 'ddpm', group: 'Standard' },
|
|
{ label: 'DPM++ 2S', value: 'dpmpp_2s', group: 'Standard' },
|
|
{ label: 'DPM++ 2M', value: 'dpmpp_2m', group: 'Standard' },
|
|
{ label: 'Heun', value: 'heun', group: 'Standard' },
|
|
{ label: 'KDPM 2', value: 'kdpm_2', group: 'Standard' },
|
|
{ label: 'LMS', value: 'lms', group: 'Standard' },
|
|
{ label: 'PNDM', value: 'pndm', group: 'Standard' },
|
|
{ label: 'UniPC', value: 'unipc', group: 'Standard' },
|
|
{ label: 'Euler Karras', value: 'euler_k', group: 'Karras' },
|
|
{ label: 'DPM++ 2S Karras', value: 'dpmpp_2s_k', group: 'Karras' },
|
|
{ label: 'DPM++ 2M Karras', value: 'dpmpp_2m_k', group: 'Karras' },
|
|
{ label: 'Heun Karras', value: 'heun_k', group: 'Karras' },
|
|
{ label: 'LMS Karras', value: 'lms_k', group: 'Karras' },
|
|
{ label: 'Euler Ancestral', value: 'euler_a', group: 'Ancestral' },
|
|
{ label: 'KDPM 2 Ancestral', value: 'kdpm_2_a', group: 'Ancestral' },
|
|
];
|
|
|
|
// zod needs the array to be `as const` to infer the type correctly
|
|
export const SCHEDULER_NAMES_AS_CONST = [
|
|
'euler',
|
|
'deis',
|
|
'ddim',
|
|
'ddpm',
|
|
'dpmpp_2s',
|
|
'dpmpp_2m',
|
|
'heun',
|
|
'kdpm_2',
|
|
'lms',
|
|
'pndm',
|
|
'unipc',
|
|
'euler_k',
|
|
'dpmpp_2s_k',
|
|
'dpmpp_2m_k',
|
|
'heun_k',
|
|
'lms_k',
|
|
'euler_a',
|
|
'kdpm_2_a',
|
|
] as const;
|
|
|
|
export const SCHEDULER_NAMES = [...SCHEDULER_NAMES_AS_CONST];
|
|
|
|
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;
|