feat(constants): add 12 new CivitAI base models from API, sync JS/Python abbreviations and categories

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",
             ],
         }
 

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",
     ]
 )

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",
+    GROK: "Grok",
-    PONY_V7: "Pony V7",
+    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.ACE_AUDIO, BASE_MODELS.BOOGU, BASE_MODELS.ERNIE, BASE_MODELS.ERNIE_TURBO,
-        BASE_MODELS.KREA_2,
+        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
     ]
 };