alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-07-05 17:01:16 -03:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Will Miao
2026-07-05 11:44:53 +08:00
parent 8fb00998a7
commit 87db23825f
3 changed files with 71 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
py/services/civitai_base_model_service.py
View File

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

14
py/utils/constants.py
View File

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

56
static/js/utils/constants.js
View File

@@ -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
]
};