@lobehub/chat
Version:
Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.
49 lines (42 loc) • 1.85 kB
text/typescript
/**
* ComfyUI Model Registry - Linus-style simple design
* Interface shared, registries split for maintainability
*/
import { FLUX_MODEL_REGISTRY } from './fluxModelRegistry';
import { SD_MODEL_REGISTRY } from './sdModelRegistry';
export interface ModelConfig {
modelFamily: string;
priority: number;
recommendedDtype?: 'default' | 'fp8_e4m3fn' | 'fp8_e4m3fn_fast' | 'fp8_e5m2';
variant: string;
}
// ===================================================================
// Combined Model Registry - FLUX + SD families
// ===================================================================
export const MODEL_REGISTRY: Record<string, ModelConfig> = {
...FLUX_MODEL_REGISTRY,
...SD_MODEL_REGISTRY,
};
/**
* Model ID to Variant mapping
* Maps actual frontend model IDs to their corresponding variants in registry
* Based on src/config/aiModels/comfyui.ts definitions
*/
/* eslint-disable sort-keys-fix/sort-keys-fix */
export const MODEL_ID_VARIANT_MAP: Record<string, string> = {
// FLUX models
'flux-schnell': 'schnell', // comfyui/flux-schnell
'flux-dev': 'dev', // comfyui/flux-dev
'flux-krea-dev': 'krea', // comfyui/flux-krea-dev
'flux-kontext-dev': 'kontext', // comfyui/flux-kontext-dev
// SD3 models
'stable-diffusion-35': 'sd35', // comfyui/stable-diffusion-35
'stable-diffusion-35-inclclip': 'sd35-inclclip', // comfyui/stable-diffusion-35-inclclip
// SD1/SDXL models
'stable-diffusion-15': 'sd15-t2i', // comfyui/stable-diffusion-15
'stable-diffusion-xl': 'sdxl-t2i', // comfyui/stable-diffusion-xl
'stable-diffusion-refiner': 'sdxl-i2i', // comfyui/stable-diffusion-refiner
'stable-diffusion-custom': 'custom-sd', // comfyui/stable-diffusion-custom
'stable-diffusion-custom-refiner': 'custom-sd', // comfyui/stable-diffusion-custom-refiner
};
/* eslint-enable sort-keys-fix/sort-keys-fix */