diff --git a/config_example.json b/config_example.json index 43df4663..44ee3dda 100644 --- a/config_example.json +++ b/config_example.json @@ -74,6 +74,7 @@ // "sk-xxxxxxxxxxxxxxxxxxxxxxxx2", // "sk-xxxxxxxxxxxxxxxxxxxxxxxx3" // ], + // "rename_model": "GPT-4o-mini", //指定默认命名模型 // 自定义OpenAI API Base // "openai_api_base": "https://api.openai.com", // 自定义使用代理(请替换代理URL) diff --git a/modules/config.py b/modules/config.py index b0945fff..d95230e9 100644 --- a/modules/config.py +++ b/modules/config.py @@ -335,6 +335,18 @@ def update_doc_config(two_column_pdf): share = config.get("share", False) autobrowser = config.get("autobrowser", True) +#设置默认命名model +rename_model = config.get("rename_model", None) +try: + if rename_model is not None: + if rename_model in presets.MODELS: + presets.RENAME_MODEL = presets.MODELS.index(rename_model) + else: + presets.RENAME_MODEL = presets.MODELS.index(next((k for k, v in presets.MODEL_METADATA.items() if v.get("model_name") == rename_model), None)) + logging.info("默认命名模型设置为了:" + str(presets.MODELS[presets.RENAME_MODEL])) +except ValueError: + logging.error("你填写的默认命名模型" + rename_model + "不存在!请从下面的列表中挑一个填写:" + str(presets.MODELS)) + # avatar bot_avatar = config.get("bot_avatar", "default") user_avatar = config.get("user_avatar", "default") diff --git a/modules/models/OpenAIVision.py b/modules/models/OpenAIVision.py index 5e0bc827..6d594e53 100644 --- a/modules/models/OpenAIVision.py +++ b/modules/models/OpenAIVision.py @@ -275,7 +275,7 @@ def _single_query_at_once(self, history, temperature=1.0): "temperature": f"{temperature}", } payload = { - "model": self.model_name, + "model": RENAME_MODEL if RENAME_MODEL is not None else self.model_name, "messages": history, } diff --git a/modules/presets.py b/modules/presets.py index deda2237..31cb2ece 100644 --- a/modules/presets.py +++ b/modules/presets.py @@ -512,6 +512,8 @@ DEFAULT_MODEL = 0 +RENAME_MODEL = 0 + os.makedirs("models", exist_ok=True) os.makedirs("lora", exist_ok=True) os.makedirs("history", exist_ok=True)