parent
80ecc08044
commit
7217e3391d
|
|
@ -64,7 +64,18 @@ llm_model_dict = {
|
|||
"local_model_path": None,
|
||||
"provides": "ChatGLM"
|
||||
},
|
||||
|
||||
"chatglm2-6b-int4": {
|
||||
"name": "chatglm2-6b-int4",
|
||||
"pretrained_model_name": "THUDM/chatglm2-6b-int4",
|
||||
"local_model_path": None,
|
||||
"provides": "ChatGLM"
|
||||
},
|
||||
"chatglm2-6b-int8": {
|
||||
"name": "chatglm2-6b-int8",
|
||||
"pretrained_model_name": "THUDM/chatglm2-6b-int8",
|
||||
"local_model_path": None,
|
||||
"provides": "ChatGLM"
|
||||
},
|
||||
"chatyuan": {
|
||||
"name": "chatyuan",
|
||||
"pretrained_model_name": "ClueAI/ChatYuan-large-v2",
|
||||
|
|
@ -141,7 +152,7 @@ PROMPT_TEMPLATE = """已知信息:
|
|||
|
||||
根据上述已知信息,简洁和专业的来回答用户的问题。如果无法从中得到答案,请说 “根据已知信息无法回答该问题” 或 “没有提供足够的相关信息”,不允许在答案中添加编造成分,答案请使用中文。 问题是:{question}"""
|
||||
|
||||
# 缓存知识库数量
|
||||
# 缓存知识库数量,如果是ChatGLM2,ChatGLM2-int4,ChatGLM2-int8模型若检索效果不好可以调成’10’
|
||||
CACHED_VS_NUM = 1
|
||||
|
||||
# 文本分句长度
|
||||
|
|
|
|||
Loading…
Reference in New Issue