waveshaker commited on
Commit
44c2fec
·
verified ·
1 Parent(s): 739f2bd

Update model_config.py

Browse files
Files changed (1) hide show
  1. 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","moss"]
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 = {