提交 49e47231 作者: imClumsyPanda

update model_config.py

上级 7d837078
...@@ -40,9 +40,15 @@ llm_model_dict = { ...@@ -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 = { ...@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论