diff --git a/static/js/components/shared/ModelMetadata.js b/static/js/components/shared/ModelMetadata.js index a8dc4e7d..960ffa7d 100644 --- a/static/js/components/shared/ModelMetadata.js +++ b/static/js/components/shared/ModelMetadata.js @@ -183,7 +183,7 @@ export function setupBaseModelEditing(filePath) { 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.HIDREAM, - BASE_MODELS.UNKNOWN + BASE_MODELS.QWEN, BASE_MODELS.UNKNOWN ] }; @@ -426,4 +426,4 @@ export function setupFileNameEditing(filePath) { fileNameWrapper.classList.remove('editing'); editBtn.classList.remove('visible'); } -} \ No newline at end of file +} diff --git a/static/js/utils/constants.js b/static/js/utils/constants.js index 53249ccb..90c91d05 100644 --- a/static/js/utils/constants.js +++ b/static/js/utils/constants.js @@ -35,6 +35,7 @@ export const BASE_MODELS = { ILLUSTRIOUS: "Illustrious", PONY: "Pony", HIDREAM: "HiDream", + QWEN: "Qwen", // Video models SVD: "SVD", @@ -93,6 +94,7 @@ export const BASE_MODEL_CLASSES = { [BASE_MODELS.ILLUSTRIOUS]: "il", [BASE_MODELS.PONY]: "pony", [BASE_MODELS.HIDREAM]: "hidream", + [BASE_MODELS.QWEN]: "qwen", // Default [BASE_MODELS.UNKNOWN]: "unknown" @@ -206,4 +208,4 @@ export const NODE_TYPE_ICONS = { }; // Default ComfyUI node color when bgcolor is null -export const DEFAULT_NODE_COLOR = "#353535"; \ No newline at end of file +export const DEFAULT_NODE_COLOR = "#353535";