import { 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 REFINER_BASE_MODELS: BaseModelType[] = ['sdxl-refiner'];