From 87db23825ff15cd5688fc2cb3dbf337f2942298d Mon Sep 17 00:00:00 2001 From: Will Miao Date: Sun, 5 Jul 2026 11:44:53 +0800 Subject: [PATCH] feat(constants): add 12 new CivitAI base models from API, sync JS/Python abbreviations and categories --- py/services/civitai_base_model_service.py | 24 ++++++++++ py/utils/constants.py | 14 +++++- static/js/utils/constants.js | 56 ++++++++++++++--------- 3 files changed, 71 insertions(+), 23 deletions(-) diff --git a/py/services/civitai_base_model_service.py b/py/services/civitai_base_model_service.py index 4bce7780..1cd61e50 100644 --- a/py/services/civitai_base_model_service.py +++ b/py/services/civitai_base_model_service.py @@ -213,6 +213,18 @@ class CivitaiBaseModelService: "wan video 2.2 i2v-a14b": "WAN", "wan video 2.5 t2v": "WAN", "wan video 2.5 i2v": "WAN", + "wan video 2.7": "WAN", + "wan image 2.7": "WI27", + "ace audio": "ACE", + "boogu": "BOOG", + "grok": "GROK", + "happyhorse": "HAPP", + "hidream-o1": "HIO1", + "lens": "LENS", + "mai": "MAI", + "upscaler": "UPSC", + "ideogram 4.0": "ID40", + "qwen 2": "QWN2", } if lower_name in special_cases: @@ -392,6 +404,7 @@ class CivitaiBaseModelService: "LTXV2", "LTXV 2.3", "CogVideoX", + "HappyHorse", "Mochi", "Hunyuan Video", "Wan Video", @@ -404,15 +417,25 @@ class CivitaiBaseModelService: "Wan Video 2.2 I2V-A14B", "Wan Video 2.5 T2V", "Wan Video 2.5 I2V", + "Wan Image 2.7", + "Wan Video 2.7", ], "Other Models": [ + "ACE Audio", "Illustrious", "Pony", "Pony V7", + "Boogu", "HiDream", + "HiDream-O1", + "Ideogram 4.0", "Qwen", + "Qwen 2", "AuraFlow", "Chroma", + "Grok", + "Lens", + "MAI", "ZImageTurbo", "ZImageBase", "PixArt a", @@ -426,6 +449,7 @@ class CivitaiBaseModelService: "Ernie Turbo", "Nucleus", "Krea 2", + "Upscaler", ], } diff --git a/py/utils/constants.py b/py/utils/constants.py index ec569fab..99fcc5d9 100644 --- a/py/utils/constants.py +++ b/py/utils/constants.py @@ -226,9 +226,21 @@ SUPPORTED_DOWNLOAD_SKIP_BASE_MODELS = frozenset( "Wan Video 2.5 I2V", "Hunyuan Video", "Anima", + "ACE Audio", + "Boogu", "Ernie", "Ernie Turbo", - "Nucleus", + "Grok", + "HappyHorse", + "HiDream-O1", + "Ideogram 4.0", "Krea 2", + "Lens", + "MAI", + "Nucleus", + "Qwen 2", + "Upscaler", + "Wan Image 2.7", + "Wan Video 2.7", ] ) diff --git a/static/js/utils/constants.js b/static/js/utils/constants.js index 33da9d6e..bb91c52d 100644 --- a/static/js/utils/constants.js +++ b/static/js/utils/constants.js @@ -66,11 +66,23 @@ export const BASE_MODELS = { HUNYUAN_VIDEO: "Hunyuan Video", // Other models ANIMA: "Anima", + ACE_AUDIO: "ACE Audio", + BOOGU: "Boogu", ERNIE: "Ernie", ERNIE_TURBO: "Ernie Turbo", - NUCLEUS: "Nucleus", - PONY_V7: "Pony V7", + GROK: "Grok", + HAPPY_HORSE: "HappyHorse", + HIDREAM_O1: "HiDream-O1", + IDEOGRAM_4_0: "Ideogram 4.0", KREA_2: "Krea 2", + LENS: "Lens", + PONY_V7: "Pony V7", + MAI: "MAI", + NUCLEUS: "Nucleus", + QWEN_2: "Qwen 2", + UPSCALER: "Upscaler", + WAN_IMAGE_2_7: "Wan Image 2.7", + WAN_VIDEO_2_7: "Wan Video 2.7", // Default UNKNOWN: "Other" }; @@ -143,22 +155,6 @@ export const BASE_MODEL_ABBREVIATIONS = { [BASE_MODELS.FLUX_2_KLEIN_4B]: 'FK4', [BASE_MODELS.FLUX_2_KLEIN_4B_BASE]: 'FK4B', - // Other diffusion models - [BASE_MODELS.AURAFLOW]: 'AF', - [BASE_MODELS.CHROMA]: 'CHR', - [BASE_MODELS.PIXART_A]: 'PXA', - [BASE_MODELS.PIXART_E]: 'PXE', - [BASE_MODELS.HUNYUAN_1]: 'HY', - [BASE_MODELS.LUMINA]: 'L', - [BASE_MODELS.KOLORS]: 'KLR', - [BASE_MODELS.NOOBAI]: 'NAI', - [BASE_MODELS.ILLUSTRIOUS]: 'IL', - [BASE_MODELS.PONY]: 'PONY', - [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', @@ -195,10 +191,22 @@ export const BASE_MODEL_ABBREVIATIONS = { [BASE_MODELS.ZIMAGE_TURBO]: 'ZIT', [BASE_MODELS.ZIMAGE_BASE]: 'ZIB', [BASE_MODELS.ANIMA]: 'ANI', + [BASE_MODELS.ACE_AUDIO]: 'ACE', + [BASE_MODELS.BOOGU]: 'BOOG', [BASE_MODELS.ERNIE]: 'ERNI', [BASE_MODELS.ERNIE_TURBO]: 'ETRB', - [BASE_MODELS.NUCLEUS]: 'NUCL', + [BASE_MODELS.GROK]: 'GROK', + [BASE_MODELS.HAPPY_HORSE]: 'HAPP', + [BASE_MODELS.HIDREAM_O1]: 'HIO1', + [BASE_MODELS.IDEOGRAM_4_0]: 'ID40', [BASE_MODELS.KREA_2]: 'KR2', + [BASE_MODELS.LENS]: 'LENS', + [BASE_MODELS.MAI]: 'MAI', + [BASE_MODELS.NUCLEUS]: 'NUCL', + [BASE_MODELS.QWEN_2]: 'QWN2', + [BASE_MODELS.UPSCALER]: 'UPSC', + [BASE_MODELS.WAN_IMAGE_2_7]: 'WI27', + [BASE_MODELS.WAN_VIDEO_2_7]: 'WAN', // Default [BASE_MODELS.UNKNOWN]: 'OTH' @@ -394,7 +402,9 @@ export const BASE_MODEL_CATEGORIES = { 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, BASE_MODELS.WAN_VIDEO_2_5_T2V, - BASE_MODELS.WAN_VIDEO_2_5_I2V + BASE_MODELS.WAN_VIDEO_2_5_I2V, + BASE_MODELS.HAPPY_HORSE, + BASE_MODELS.WAN_IMAGE_2_7, BASE_MODELS.WAN_VIDEO_2_7 ], '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': [ @@ -402,8 +412,10 @@ export const BASE_MODEL_CATEGORIES = { 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.ANIMA, - BASE_MODELS.ERNIE, BASE_MODELS.ERNIE_TURBO, BASE_MODELS.NUCLEUS, - BASE_MODELS.KREA_2, + BASE_MODELS.ACE_AUDIO, BASE_MODELS.BOOGU, BASE_MODELS.ERNIE, BASE_MODELS.ERNIE_TURBO, + BASE_MODELS.GROK, BASE_MODELS.HIDREAM_O1, BASE_MODELS.IDEOGRAM_4_0, + BASE_MODELS.LENS, BASE_MODELS.MAI, BASE_MODELS.NUCLEUS, + BASE_MODELS.QWEN_2, BASE_MODELS.KREA_2, BASE_MODELS.UPSCALER, BASE_MODELS.UNKNOWN ] };