Spaces:
Running on Zero
Running on Zero
Update model_config.py
Browse files- model_config.py +3 -4
model_config.py
CHANGED
|
@@ -5,7 +5,7 @@ import os
|
|
| 5 |
# 设为 False 则加载全部模型 (生产模式)
|
| 6 |
# ==========================================================================
|
| 7 |
DEV_MODE = True
|
| 8 |
-
DEV_TARGET_MODEL = ["pony"
|
| 9 |
|
| 10 |
# ==========================================================================
|
| 11 |
# 🛠️ 环境开关 (动态控制 Gradio UI 是否开启)
|
|
@@ -231,12 +231,11 @@ if DEV_MODE:
|
|
| 231 |
if k in DEV_TARGET_MODEL
|
| 232 |
}
|
| 233 |
|
| 234 |
-
# 如果处于开发模式,提取目标模型列表;如果生产模式,提取所有已配置的模型 Key
|
| 235 |
if DEV_MODE:
|
| 236 |
-
#
|
| 237 |
display_models = ", ".join([f'"{m.upper()}"' for m in DEV_TARGET_MODEL])
|
| 238 |
else:
|
| 239 |
-
# 提取所有模型 Key,转大写
|
| 240 |
display_models = ", ".join([f'"{m.upper()}"' for m in MODEL_CONFIG["json_keys"]["root_mapping"].keys()])
|
| 241 |
|
| 242 |
CORE_STATUS = {
|
|
|
|
| 5 |
# 设为 False 则加载全部模型 (生产模式)
|
| 6 |
# ==========================================================================
|
| 7 |
DEV_MODE = True
|
| 8 |
+
DEV_TARGET_MODEL = ["pony","moss"]
|
| 9 |
|
| 10 |
# ==========================================================================
|
| 11 |
# 🛠️ 环境开关 (动态控制 Gradio UI 是否开启)
|
|
|
|
| 231 |
if k in DEV_TARGET_MODEL
|
| 232 |
}
|
| 233 |
|
|
|
|
| 234 |
if DEV_MODE:
|
| 235 |
+
# 如果开启 DEV_MODE,就把目标列表转大写并加上双引号
|
| 236 |
display_models = ", ".join([f'"{m.upper()}"' for m in DEV_TARGET_MODEL])
|
| 237 |
else:
|
| 238 |
+
# 如果是生产模式,提取所有模型的 Key,转大写并加上双引号
|
| 239 |
display_models = ", ".join([f'"{m.upper()}"' for m in MODEL_CONFIG["json_keys"]["root_mapping"].keys()])
|
| 240 |
|
| 241 |
CORE_STATUS = {
|