From 7803bd542d12e70699cb229f5349c496e1a7d8f2 Mon Sep 17 00:00:00 2001
From: Will Miao <willmiao.dev@gmail.com>
Date: Thu, 7 May 2026 10:49:01 +0800
Subject: [PATCH] feat(base-models): add Ernie, Ernie Turbo, Nucleus base model
 types (#922)

- Ernie & Anima: auto-fetched via CivitaiBaseModelService from Civitai API
- Ernie Turbo & Nucleus: pre-added as hardcoded constants (not yet in Civitai API)
- Added abbreviations (ERNI, ETRB, NUCL) and category entries across all layers
---
 py/services/civitai_base_model_service.py | 6 ++++++
 py/utils/constants.py                     | 3 +++
 static/js/utils/constants.js              | 7 +++++++
 3 files changed, 16 insertions(+)

diff --git a/py/services/civitai_base_model_service.py b/py/services/civitai_base_model_service.py
index dc213ded..e0dc89b9 100644
--- a/py/services/civitai_base_model_service.py
+++ b/py/services/civitai_base_model_service.py
@@ -193,6 +193,9 @@ class CivitaiBaseModelService:
             "zimageturbo": "ZIT",
             "zimagebase": "ZIB",
             "anima": "ANI",
+            "ernie": "ERNI",
+            "ernie turbo": "ETRB",
+            "nucleus": "NUCL",
             "svd": "SVD",
             "ltxv": "LTXV",
             "ltxv2": "LTV2",
@@ -418,6 +421,9 @@ class CivitaiBaseModelService:
                 "Kolors",
                 "NoobAI",
                 "Anima",
+                "Ernie",
+                "Ernie Turbo",
+                "Nucleus",
             ],
         }
 
diff --git a/py/utils/constants.py b/py/utils/constants.py
index 641637be..1121ad91 100644
--- a/py/utils/constants.py
+++ b/py/utils/constants.py
@@ -178,5 +178,8 @@ SUPPORTED_DOWNLOAD_SKIP_BASE_MODELS = frozenset(
         "Wan Video 2.5 I2V",
         "Hunyuan Video",
         "Anima",
+        "Ernie",
+        "Ernie Turbo",
+        "Nucleus",
     ]
 )
diff --git a/static/js/utils/constants.js b/static/js/utils/constants.js
index 6630d5fe..ceede86c 100644
--- a/static/js/utils/constants.js
+++ b/static/js/utils/constants.js
@@ -66,6 +66,9 @@ export const BASE_MODELS = {
     HUNYUAN_VIDEO: "Hunyuan Video",
     // Other models
     ANIMA: "Anima",
+    ERNIE: "Ernie",
+    ERNIE_TURBO: "Ernie Turbo",
+    NUCLEUS: "Nucleus",
     PONY_V7: "Pony V7",
     // Default
     UNKNOWN: "Other"
@@ -191,6 +194,9 @@ export const BASE_MODEL_ABBREVIATIONS = {
     [BASE_MODELS.ZIMAGE_TURBO]: 'ZIT',
     [BASE_MODELS.ZIMAGE_BASE]: 'ZIB',
     [BASE_MODELS.ANIMA]: 'ANI',
+    [BASE_MODELS.ERNIE]: 'ERNI',
+    [BASE_MODELS.ERNIE_TURBO]: 'ETRB',
+    [BASE_MODELS.NUCLEUS]: 'NUCL',
 
     // Default
     [BASE_MODELS.UNKNOWN]: 'OTH'
@@ -394,6 +400,7 @@ 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.UNKNOWN
     ]
 };