diff --git a/static/js/utils/constants.js b/static/js/utils/constants.js index 8870a4c6..328ceb9b 100644 --- a/static/js/utils/constants.js +++ b/static/js/utils/constants.js @@ -27,6 +27,10 @@ export const BASE_MODELS = { FLUX_1_KREA: "Flux.1 Krea", FLUX_1_KONTEXT: "Flux.1 Kontext", 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", CHROMA: "Chroma", PIXART_A: "PixArt a", @@ -40,10 +44,12 @@ export const BASE_MODELS = { HIDREAM: "HiDream", QWEN: "Qwen", ZIMAGE_TURBO: "ZImageTurbo", - + ZIMAGE_BASE: "ZImageBase", + // Video models SVD: "SVD", LTXV: "LTXV", + LTXV2: "LTXV2", WAN_VIDEO: "Wan Video", WAN_VIDEO_1_3B_T2V: "Wan Video 1.3B 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_KONTEXT]: 'F1KX', [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 [BASE_MODELS.AURAFLOW]: 'AF', @@ -135,10 +145,12 @@ export const BASE_MODEL_ABBREVIATIONS = { [BASE_MODELS.HIDREAM]: 'HID', [BASE_MODELS.QWEN]: 'QWEN', [BASE_MODELS.ZIMAGE_TURBO]: 'ZIT', + [BASE_MODELS.ZIMAGE_BASE]: 'ZIB', // Video models [BASE_MODELS.SVD]: 'SVD', [BASE_MODELS.LTXV]: 'LTXV', + [BASE_MODELS.LTXV2]: 'LTV2', [BASE_MODELS.WAN_VIDEO]: 'WAN', [BASE_MODELS.WAN_VIDEO_1_3B_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], 'SDXL': [BASE_MODELS.SDXL, BASE_MODELS.SDXL_LIGHTNING, BASE_MODELS.SDXL_HYPER], '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_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, 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': [ 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.LUMINA, BASE_MODELS.KOLORS, BASE_MODELS.NOOBAI, BASE_MODELS.UNKNOWN