更多模型切换

2023-04-17 21:23:03 +08:00
parent 03ba072c16
commit 9bd8511ba4
5 changed files with 166 additions and 141 deletions
--- a/config.py
+++ b/config.py
@@ -46,14 +46,12 @@ WEB_PORT = -1
 MAX_RETRY = 2

 # OpenAI模型选择是（gpt4现在只对申请成功的人开放）
-LLM_MODEL = "gpt-3.5-turbo" # 可选 "chatglm", "tgui:anymodel@localhost:7865"
+LLM_MODEL = "gpt-3.5-turbo" # 可选 "chatglm"
+AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "chatglm", "gpt-4", "api2d-gpt-4", "api2d-gpt-3.5-turbo"]

 # 本地LLM模型如ChatGLM的执行方式 CPU/GPU
 LOCAL_MODEL_DEVICE = "cpu" # 可选 "cuda"

-# OpenAI的API_URL
-API_URL = "https://api.openai.com/v1/chat/completions"
-
 # 设置gradio的并行线程数（不需要修改）
 CONCURRENT_COUNT = 100