mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
feat: add new Flux Klein models, ZImageBase, and LTXV2 to constants, see #792
- Add Flux.2 Klein 9B, 9B-base, 4B, and 4B-base models to BASE_MODELS, BASE_MODEL_ABBREVIATIONS, and Flux Models category - Include ZImageBase model and its abbreviation - Add LTXV2 video model to BASE_MODELS, BASE_MODEL_ABBREVIATIONS, and Video Models category - Update model categories to reflect new additions
This commit is contained in:
@@ -27,6 +27,10 @@ export const BASE_MODELS = {
|
|||||||
FLUX_1_KREA: "Flux.1 Krea",
|
FLUX_1_KREA: "Flux.1 Krea",
|
||||||
FLUX_1_KONTEXT: "Flux.1 Kontext",
|
FLUX_1_KONTEXT: "Flux.1 Kontext",
|
||||||
FLUX_2_D: "Flux.2 D",
|
FLUX_2_D: "Flux.2 D",
|
||||||
|
FLUX_2_KLEIN_9B: "Flux.2 Klein 9B",
|
||||||
|
FLUX_2_KLEIN_9B_BASE: "Flux.2 Klein 9B-base",
|
||||||
|
FLUX_2_KLEIN_4B: "Flux.2 Klein 4B",
|
||||||
|
FLUX_2_KLEIN_4B_BASE: "Flux.2 Klein 4B-base",
|
||||||
AURAFLOW: "AuraFlow",
|
AURAFLOW: "AuraFlow",
|
||||||
CHROMA: "Chroma",
|
CHROMA: "Chroma",
|
||||||
PIXART_A: "PixArt a",
|
PIXART_A: "PixArt a",
|
||||||
@@ -40,10 +44,12 @@ export const BASE_MODELS = {
|
|||||||
HIDREAM: "HiDream",
|
HIDREAM: "HiDream",
|
||||||
QWEN: "Qwen",
|
QWEN: "Qwen",
|
||||||
ZIMAGE_TURBO: "ZImageTurbo",
|
ZIMAGE_TURBO: "ZImageTurbo",
|
||||||
|
ZIMAGE_BASE: "ZImageBase",
|
||||||
|
|
||||||
// Video models
|
// Video models
|
||||||
SVD: "SVD",
|
SVD: "SVD",
|
||||||
LTXV: "LTXV",
|
LTXV: "LTXV",
|
||||||
|
LTXV2: "LTXV2",
|
||||||
WAN_VIDEO: "Wan Video",
|
WAN_VIDEO: "Wan Video",
|
||||||
WAN_VIDEO_1_3B_T2V: "Wan Video 1.3B t2v",
|
WAN_VIDEO_1_3B_T2V: "Wan Video 1.3B t2v",
|
||||||
WAN_VIDEO_14B_T2V: "Wan Video 14B t2v",
|
WAN_VIDEO_14B_T2V: "Wan Video 14B t2v",
|
||||||
@@ -120,6 +126,10 @@ export const BASE_MODEL_ABBREVIATIONS = {
|
|||||||
[BASE_MODELS.FLUX_1_KREA]: 'F1KR',
|
[BASE_MODELS.FLUX_1_KREA]: 'F1KR',
|
||||||
[BASE_MODELS.FLUX_1_KONTEXT]: 'F1KX',
|
[BASE_MODELS.FLUX_1_KONTEXT]: 'F1KX',
|
||||||
[BASE_MODELS.FLUX_2_D]: 'F2D',
|
[BASE_MODELS.FLUX_2_D]: 'F2D',
|
||||||
|
[BASE_MODELS.FLUX_2_KLEIN_9B]: 'FK9',
|
||||||
|
[BASE_MODELS.FLUX_2_KLEIN_9B_BASE]: 'FK9B',
|
||||||
|
[BASE_MODELS.FLUX_2_KLEIN_4B]: 'FK4',
|
||||||
|
[BASE_MODELS.FLUX_2_KLEIN_4B_BASE]: 'FK4B',
|
||||||
|
|
||||||
// Other diffusion models
|
// Other diffusion models
|
||||||
[BASE_MODELS.AURAFLOW]: 'AF',
|
[BASE_MODELS.AURAFLOW]: 'AF',
|
||||||
@@ -135,10 +145,12 @@ export const BASE_MODEL_ABBREVIATIONS = {
|
|||||||
[BASE_MODELS.HIDREAM]: 'HID',
|
[BASE_MODELS.HIDREAM]: 'HID',
|
||||||
[BASE_MODELS.QWEN]: 'QWEN',
|
[BASE_MODELS.QWEN]: 'QWEN',
|
||||||
[BASE_MODELS.ZIMAGE_TURBO]: 'ZIT',
|
[BASE_MODELS.ZIMAGE_TURBO]: 'ZIT',
|
||||||
|
[BASE_MODELS.ZIMAGE_BASE]: 'ZIB',
|
||||||
|
|
||||||
// Video models
|
// Video models
|
||||||
[BASE_MODELS.SVD]: 'SVD',
|
[BASE_MODELS.SVD]: 'SVD',
|
||||||
[BASE_MODELS.LTXV]: 'LTXV',
|
[BASE_MODELS.LTXV]: 'LTXV',
|
||||||
|
[BASE_MODELS.LTXV2]: 'LTV2',
|
||||||
[BASE_MODELS.WAN_VIDEO]: 'WAN',
|
[BASE_MODELS.WAN_VIDEO]: 'WAN',
|
||||||
[BASE_MODELS.WAN_VIDEO_1_3B_T2V]: 'WAN',
|
[BASE_MODELS.WAN_VIDEO_1_3B_T2V]: 'WAN',
|
||||||
[BASE_MODELS.WAN_VIDEO_14B_T2V]: 'WAN',
|
[BASE_MODELS.WAN_VIDEO_14B_T2V]: 'WAN',
|
||||||
@@ -328,16 +340,16 @@ export const BASE_MODEL_CATEGORIES = {
|
|||||||
'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': [
|
'Video Models': [
|
||||||
BASE_MODELS.SVD, BASE_MODELS.LTXV, BASE_MODELS.HUNYUAN_VIDEO, BASE_MODELS.WAN_VIDEO,
|
BASE_MODELS.SVD, BASE_MODELS.LTXV, BASE_MODELS.LTXV2, 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_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_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_TI2V_5B, BASE_MODELS.WAN_VIDEO_2_2_T2V_A14B,
|
||||||
BASE_MODELS.WAN_VIDEO_2_2_I2V_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, BASE_MODELS.FLUX_2_D],
|
'Flux Models': [BASE_MODELS.FLUX_1_D, BASE_MODELS.FLUX_1_S, BASE_MODELS.FLUX_1_KONTEXT, BASE_MODELS.FLUX_1_KREA, BASE_MODELS.FLUX_2_D, BASE_MODELS.FLUX_2_KLEIN_9B, BASE_MODELS.FLUX_2_KLEIN_9B_BASE, BASE_MODELS.FLUX_2_KLEIN_4B, BASE_MODELS.FLUX_2_KLEIN_4B_BASE],
|
||||||
'Other Models': [
|
'Other Models': [
|
||||||
BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM,
|
BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM,
|
||||||
BASE_MODELS.QWEN, BASE_MODELS.AURAFLOW, BASE_MODELS.CHROMA, BASE_MODELS.ZIMAGE_TURBO,
|
BASE_MODELS.QWEN, BASE_MODELS.AURAFLOW, BASE_MODELS.CHROMA, BASE_MODELS.ZIMAGE_TURBO, BASE_MODELS.ZIMAGE_BASE,
|
||||||
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.UNKNOWN
|
BASE_MODELS.UNKNOWN
|
||||||
|
|||||||
Reference in New Issue
Block a user