CC/gpt_academic
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

logging -> loguru: final stage

Browse Source
This commit is contained in:
binary-husky
2024-09-15 15:51:51 +00:00
parent bbf9e9f868
commit 2f343179a2
55 changed files with 237 additions and 529 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
request_llms/bridge_all.py
View File

@@ -1022,7 +1022,7 @@ for model in [m for m in AVAIL_LLM_MODELS if m.startswith("one-api-")]:
# 如果是已知模型，则尝试获取其信息
original_model_info = model_info.get(origin_model_name.replace("one-api-", "", 1), None)
except:
print(f"one-api模型 {model} 的 max_token 配置不是整数，请检查配置文件。")
logger.error(f"one-api模型 {model} 的 max_token 配置不是整数，请检查配置文件。")
continue
this_model_info = {
"fn_with_ui": chatgpt_ui,
@@ -1053,7 +1053,7 @@ for model in [m for m in AVAIL_LLM_MODELS if m.startswith("vllm-")]:
try:
_, max_token_tmp = read_one_api_model_name(model)
except:
print(f"vllm模型 {model} 的 max_token 配置不是整数，请检查配置文件。")
logger.error(f"vllm模型 {model} 的 max_token 配置不是整数，请检查配置文件。")
continue
model_info.update({
model: {
@@ -1080,7 +1080,7 @@ for model in [m for m in AVAIL_LLM_MODELS if m.startswith("ollama-")]:
try:
_, max_token_tmp = read_one_api_model_name(model)
except:
print(f"ollama模型 {model} 的 max_token 配置不是整数，请检查配置文件。")
logger.error(f"ollama模型 {model} 的 max_token 配置不是整数，请检查配置文件。")
continue
model_info.update({
model: {