diff --git a/static/js/components/checkpointModal/ModelMetadata.js b/static/js/components/checkpointModal/ModelMetadata.js index 18167f3b..cc3f4046 100644 --- a/static/js/components/checkpointModal/ModelMetadata.js +++ b/static/js/components/checkpointModal/ModelMetadata.js @@ -171,12 +171,13 @@ 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.WAN_VIDEO, BASE_MODELS.HUNYUAN_VIDEO], + 'Video Models': [BASE_MODELS.SVD, BASE_MODELS.LTXV, BASE_MODELS.WAN_VIDEO, BASE_MODELS.HUNYUAN_VIDEO], 'Other Models': [ BASE_MODELS.FLUX_1_D, BASE_MODELS.FLUX_1_S, 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.UNKNOWN + BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM, + BASE_MODELS.UNKNOWN ] }; diff --git a/static/js/components/loraModal/ModelMetadata.js b/static/js/components/loraModal/ModelMetadata.js index 81300150..18566e0b 100644 --- a/static/js/components/loraModal/ModelMetadata.js +++ b/static/js/components/loraModal/ModelMetadata.js @@ -173,12 +173,13 @@ 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.WAN_VIDEO, BASE_MODELS.HUNYUAN_VIDEO], + 'Video Models': [BASE_MODELS.SVD, BASE_MODELS.LTXV, BASE_MODELS.WAN_VIDEO, BASE_MODELS.HUNYUAN_VIDEO], 'Other Models': [ BASE_MODELS.FLUX_1_D, BASE_MODELS.FLUX_1_S, 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.UNKNOWN + BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM, + BASE_MODELS.UNKNOWN ] }; diff --git a/static/js/utils/constants.js b/static/js/utils/constants.js index 0711f675..248c4906 100644 --- a/static/js/utils/constants.js +++ b/static/js/utils/constants.js @@ -33,9 +33,11 @@ export const BASE_MODELS = { NOOBAI: "NoobAI", ILLUSTRIOUS: "Illustrious", PONY: "Pony", + HIDREAM: "HiDream", // Video models SVD: "SVD", + LTXV: "LTXV", WAN_VIDEO: "Wan Video", HUNYUAN_VIDEO: "Hunyuan Video", @@ -69,6 +71,7 @@ export const BASE_MODEL_CLASSES = { // Video models [BASE_MODELS.SVD]: "svd", + [BASE_MODELS.LTXV]: "ltxv", [BASE_MODELS.WAN_VIDEO]: "wan-video", [BASE_MODELS.HUNYUAN_VIDEO]: "hunyuan-video", @@ -84,6 +87,7 @@ export const BASE_MODEL_CLASSES = { [BASE_MODELS.NOOBAI]: "noobai", [BASE_MODELS.ILLUSTRIOUS]: "il", [BASE_MODELS.PONY]: "pony", + [BASE_MODELS.HIDREAM]: "hidream", // Default [BASE_MODELS.UNKNOWN]: "unknown"