weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update model_config.py

This commit is contained in:
imClumsyPanda 2023-05-19 23:16:39 +08:00
parent 7d837078e7
commit 49e47231af
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
configs/model_config.py
View File

@ -40,9 +40,15 @@ llm_model_dict = {
"local_model_path": None, "local_model_path": None,
"provides": "ChatGLM" "provides": "ChatGLM"
}, },
"chatglm-6b-int8": {
"name": "chatglm-6b-int8",
"pretrained_model_name": "THUDM/chatglm-6b-int8",
"local_model_path": None,
"provides": "ChatGLM"
},
"chatglm-6b": { "chatglm-6b": {
"name": "chatglm-6b", "name": "chatglm-6b",
"pretrained_model_name": "THUDM/chatglm-6b-int4", "pretrained_model_name": "THUDM/chatglm-6b",
"local_model_path": None, "local_model_path": None,
"provides": "ChatGLM" "provides": "ChatGLM"
}, },
@ -64,12 +70,12 @@ llm_model_dict = {
"local_model_path": None, "local_model_path": None,
"provides": None "provides": None
}, },
"chatglm-6b-int8":{ "moss": {
"name": "chatglm-6b-int8", "name": "moss",
"pretrained_model_name": "THUDM/chatglm-6b-int8", "pretrained_model_name": "fnlp/moss-moon-003-sft",
"local_model_path": None, "local_model_path": None,
"provides": "ChatGLM" "provides": "MOSSLLM"
}, }
} }
# LLM model name # LLM model name