feat: Add new Flux model 'FLUX_1_KREA' and update Video Models list for enhanced model support

This commit is contained in:
Will Miao
2025-08-28 16:24:01 +08:00
parent e90ba31784
commit 975ff2672d
2 changed files with 12 additions and 54 deletions

View File

@@ -177,13 +177,20 @@ export function setupBaseModelEditing(filePath) {
'Stable Diffusion 2.x': [BASE_MODELS.SD_2_0, BASE_MODELS.SD_2_1],
'Stable Diffusion 3.x': [BASE_MODELS.SD_3, BASE_MODELS.SD_3_5, BASE_MODELS.SD_3_5_MEDIUM, BASE_MODELS.SD_3_5_LARGE, BASE_MODELS.SD_3_5_LARGE_TURBO],
'SDXL': [BASE_MODELS.SDXL, BASE_MODELS.SDXL_LIGHTNING, BASE_MODELS.SDXL_HYPER],
'Video Models': [BASE_MODELS.SVD, BASE_MODELS.LTXV, BASE_MODELS.WAN_VIDEO, BASE_MODELS.HUNYUAN_VIDEO],
'Video Models': [
BASE_MODELS.SVD, BASE_MODELS.LTXV, BASE_MODELS.HUNYUAN_VIDEO, BASE_MODELS.WAN_VIDEO,
BASE_MODELS.WAN_VIDEO_1_3B_T2V, BASE_MODELS.WAN_VIDEO_14B_T2V,
BASE_MODELS.WAN_VIDEO_14B_I2V_480P, BASE_MODELS.WAN_VIDEO_14B_I2V_720P,
BASE_MODELS.WAN_VIDEO_2_2_TI2V_5B, BASE_MODELS.WAN_VIDEO_2_2_T2V_A14B,
BASE_MODELS.WAN_VIDEO_2_2_I2V_A14B
],
'Flux Models': [BASE_MODELS.FLUX_1_D, BASE_MODELS.FLUX_1_S, BASE_MODELS.FLUX_1_KONTEXT, BASE_MODELS.FLUX_1_KREA],
'Other Models': [
BASE_MODELS.FLUX_1_D, BASE_MODELS.FLUX_1_S, BASE_MODELS.FLUX_1_KONTEXT, BASE_MODELS.AURAFLOW,
BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM,
BASE_MODELS.QWEN, BASE_MODELS.AURAFLOW,
BASE_MODELS.PIXART_A, BASE_MODELS.PIXART_E, BASE_MODELS.HUNYUAN_1,
BASE_MODELS.LUMINA, BASE_MODELS.KOLORS, BASE_MODELS.NOOBAI,
BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM,
BASE_MODELS.QWEN, BASE_MODELS.UNKNOWN
BASE_MODELS.UNKNOWN
]
};

View File

@@ -24,6 +24,7 @@ export const BASE_MODELS = {
// Other models
FLUX_1_D: "Flux.1 D",
FLUX_1_S: "Flux.1 S",
FLUX_1_KREA: "Flux.1 Krea",
FLUX_1_KONTEXT: "Flux.1 Kontext",
AURAFLOW: "AuraFlow",
PIXART_A: "PixArt a",
@@ -53,56 +54,6 @@ export const BASE_MODELS = {
UNKNOWN: "Other"
};
// Base model display names and their corresponding class names (for styling)
export const BASE_MODEL_CLASSES = {
// Stable Diffusion 1.x models
[BASE_MODELS.SD_1_4]: "sd-1-4",
[BASE_MODELS.SD_1_5]: "sd-1-5",
[BASE_MODELS.SD_1_5_LCM]: "sd-1-5-lcm",
[BASE_MODELS.SD_1_5_HYPER]: "sd-1-5-hyper",
// Stable Diffusion 2.x models
[BASE_MODELS.SD_2_0]: "sd-2-0",
[BASE_MODELS.SD_2_1]: "sd-2-1",
// Stable Diffusion 3.x models
[BASE_MODELS.SD_3]: "sd-3",
[BASE_MODELS.SD_3_5]: "sd-3-5",
[BASE_MODELS.SD_3_5_MEDIUM]: "sd-3-5-medium",
[BASE_MODELS.SD_3_5_LARGE]: "sd-3-5-large",
[BASE_MODELS.SD_3_5_LARGE_TURBO]: "sd-3-5-large-turbo",
// SDXL models
[BASE_MODELS.SDXL]: "sdxl",
[BASE_MODELS.SDXL_LIGHTNING]: "sdxl-lightning",
[BASE_MODELS.SDXL_HYPER]: "sdxl-hyper",
// Video models
[BASE_MODELS.SVD]: "svd",
[BASE_MODELS.LTXV]: "ltxv",
[BASE_MODELS.WAN_VIDEO]: "wan-video",
[BASE_MODELS.HUNYUAN_VIDEO]: "hunyuan-video",
// Other models
[BASE_MODELS.FLUX_1_D]: "flux-d",
[BASE_MODELS.FLUX_1_S]: "flux-s",
[BASE_MODELS.FLUX_1_KONTEXT]: "flux-kontext",
[BASE_MODELS.AURAFLOW]: "auraflow",
[BASE_MODELS.PIXART_A]: "pixart-a",
[BASE_MODELS.PIXART_E]: "pixart-e",
[BASE_MODELS.HUNYUAN_1]: "hunyuan-1",
[BASE_MODELS.LUMINA]: "lumina",
[BASE_MODELS.KOLORS]: "kolors",
[BASE_MODELS.NOOBAI]: "noobai",
[BASE_MODELS.ILLUSTRIOUS]: "il",
[BASE_MODELS.PONY]: "pony",
[BASE_MODELS.HIDREAM]: "hidream",
[BASE_MODELS.QWEN]: "qwen",
// Default
[BASE_MODELS.UNKNOWN]: "unknown"
};
// Path template constants for download organization
export const DOWNLOAD_PATH_TEMPLATES = {
FLAT: {