diff --git a/static/js/utils/constants.js b/static/js/utils/constants.js index ea6c7f3f..9d36e91f 100644 --- a/static/js/utils/constants.js +++ b/static/js/utils/constants.js @@ -26,6 +26,7 @@ export const BASE_MODELS = { FLUX_1_S: "Flux.1 S", FLUX_1_KREA: "Flux.1 Krea", FLUX_1_KONTEXT: "Flux.1 Kontext", + FLUX_2_D: "Flux.2 D", AURAFLOW: "AuraFlow", CHROMA: "Chroma", PIXART_A: "PixArt a", @@ -38,6 +39,7 @@ export const BASE_MODELS = { PONY: "Pony", HIDREAM: "HiDream", QWEN: "Qwen", + ZIMAGE_TURBO: "ZImageTurbo", // Video models SVD: "SVD", @@ -89,6 +91,7 @@ export const BASE_MODEL_ABBREVIATIONS = { [BASE_MODELS.FLUX_1_S]: 'F1S', [BASE_MODELS.FLUX_1_KREA]: 'F1KR', [BASE_MODELS.FLUX_1_KONTEXT]: 'F1KX', + [BASE_MODELS.FLUX_2_D]: 'F2D', // Other diffusion models [BASE_MODELS.AURAFLOW]: 'AF', @@ -103,6 +106,7 @@ export const BASE_MODEL_ABBREVIATIONS = { [BASE_MODELS.PONY]: 'PONY', [BASE_MODELS.HIDREAM]: 'HID', [BASE_MODELS.QWEN]: 'QWEN', + [BASE_MODELS.ZIMAGE_TURBO]: 'ZIT', // Video models [BASE_MODELS.SVD]: 'SVD', @@ -302,10 +306,10 @@ export const BASE_MODEL_CATEGORIES = { 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], + '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], 'Other Models': [ BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM, - BASE_MODELS.QWEN, BASE_MODELS.AURAFLOW, BASE_MODELS.CHROMA, + BASE_MODELS.QWEN, BASE_MODELS.AURAFLOW, BASE_MODELS.CHROMA, BASE_MODELS.ZIMAGE_TURBO, BASE_MODELS.PIXART_A, BASE_MODELS.PIXART_E, BASE_MODELS.HUNYUAN_1, BASE_MODELS.LUMINA, BASE_MODELS.KOLORS, BASE_MODELS.NOOBAI, BASE_MODELS.UNKNOWN