mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-24 22:52:12 -03:00
feat: Add new Flux model 'FLUX_1_KREA' and update Video Models list for enhanced model support
This commit is contained in:
@@ -177,13 +177,20 @@ export function setupBaseModelEditing(filePath) {
|
|||||||
'Stable Diffusion 2.x': [BASE_MODELS.SD_2_0, BASE_MODELS.SD_2_1],
|
'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],
|
'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],
|
'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': [
|
'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.PIXART_A, BASE_MODELS.PIXART_E, BASE_MODELS.HUNYUAN_1,
|
||||||
BASE_MODELS.LUMINA, BASE_MODELS.KOLORS, BASE_MODELS.NOOBAI,
|
BASE_MODELS.LUMINA, BASE_MODELS.KOLORS, BASE_MODELS.NOOBAI,
|
||||||
BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM,
|
BASE_MODELS.UNKNOWN
|
||||||
BASE_MODELS.QWEN, BASE_MODELS.UNKNOWN
|
|
||||||
]
|
]
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -24,6 +24,7 @@ export const BASE_MODELS = {
|
|||||||
// Other models
|
// Other models
|
||||||
FLUX_1_D: "Flux.1 D",
|
FLUX_1_D: "Flux.1 D",
|
||||||
FLUX_1_S: "Flux.1 S",
|
FLUX_1_S: "Flux.1 S",
|
||||||
|
FLUX_1_KREA: "Flux.1 Krea",
|
||||||
FLUX_1_KONTEXT: "Flux.1 Kontext",
|
FLUX_1_KONTEXT: "Flux.1 Kontext",
|
||||||
AURAFLOW: "AuraFlow",
|
AURAFLOW: "AuraFlow",
|
||||||
PIXART_A: "PixArt a",
|
PIXART_A: "PixArt a",
|
||||||
@@ -53,56 +54,6 @@ export const BASE_MODELS = {
|
|||||||
UNKNOWN: "Other"
|
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
|
// Path template constants for download organization
|
||||||
export const DOWNLOAD_PATH_TEMPLATES = {
|
export const DOWNLOAD_PATH_TEMPLATES = {
|
||||||
FLAT: {
|
FLAT: {
|
||||||
|
|||||||
Reference in New Issue
Block a user