import type { BaseModelType } from './types';

export const ALL_BASE_MODELS: BaseModelType[] = ['sd-1', 'sd-2', 'sdxl', 'sdxl-refiner'];

export const NON_REFINER_BASE_MODELS: BaseModelType[] = ['sd-1', 'sd-2', 'sdxl'];

export const SDXL_MAIN_MODELS: BaseModelType[] = ['sdxl'];
export const NON_SDXL_MAIN_MODELS: BaseModelType[] = ['sd-1', 'sd-2'];

export const REFINER_BASE_MODELS: BaseModelType[] = ['sdxl-refiner'];