Skip to content

Commit

Permalink
feat: 新增了模型设置提示,可以在config中填写显示名称或真实模型名称 #1003
Browse files Browse the repository at this point in the history
  • Loading branch information
GaiZhenbiao committed Dec 20, 2023
1 parent 0a3d9bd commit 01aabfa
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions modules/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
import logging
import sys
import commentjson as json
import colorama

from . import shared
from . import presets
Expand Down Expand Up @@ -289,9 +290,13 @@ def update_doc_config(two_column_pdf):
# 设置默认model
default_model = config.get("default_model", "")
try:
presets.DEFAULT_MODEL = presets.MODELS.index(default_model)
if default_model in presets.MODELS:
presets.DEFAULT_MODEL = presets.MODELS.index(default_model)
else:
presets.DEFAULT_MODEL = presets.MODELS.index(next((k for k, v in presets.MODEL_METADATA.items() if v.get("model_name") == default_model), None))
logging.info("默认模型设置为了:" + str(presets.MODELS[presets.DEFAULT_MODEL]))
except ValueError:
pass
logging.error("你填写的默认模型" + default_model + "不存在!请从下面的列表中挑一个填写:" + str(presets.MODELS))

share = config.get("share", False)

Expand Down

0 comments on commit 01aabfa

Please sign in to comment.