蒲源 commited on
Commit
ae77d05
·
1 Parent(s): 47391ff

polish(pu): set moonshot-v1-32k as default model

Browse files
Files changed (2) hide show
  1. app_mqa_database.py +1 -1
  2. rag_demo.py +2 -1
app_mqa_database.py CHANGED
@@ -193,7 +193,7 @@ if __name__ == "__main__":
193
  placeholder="请您在这里输入任何关于 LightZero 的问题。",
194
  label="问题")
195
  # temperature = gr.Slider(minimum=0.0, maximum=1.0, value=0.01, step=0.01, label="温度参数")
196
- k = gr.Slider(minimum=0, maximum=7, value=3, step=1, label="检索相关文档块的数量") # readme总长度为35000左右,文段块长度为5000,因此最大值为35000/5000=7
197
  with gr.Row():
198
  gr_submit = gr.Button('提交')
199
  gr_clear = gr.Button('清除对话历史')
 
193
  placeholder="请您在这里输入任何关于 LightZero 的问题。",
194
  label="问题")
195
  # temperature = gr.Slider(minimum=0.0, maximum=1.0, value=0.01, step=0.01, label="温度参数")
196
+ k = gr.Slider(minimum=1, maximum=7, value=3, step=1, label="检索相关文档块的数量") # readme总长度为35000左右,文段块长度为5000,因此最大值为35000/5000=7
197
  with gr.Row():
198
  gr_submit = gr.Button('提交')
199
  gr_clear = gr.Button('清除对话历史')
rag_demo.py CHANGED
@@ -217,7 +217,8 @@ def execute_query_no_rag(model_name="gpt-4", temperature=0, query=""):
217
  "content": query},
218
  ]
219
  completion = client.chat.completions.create(
220
- model="moonshot-v1-128k",
 
221
  messages=messages,
222
  temperature=temperature,
223
  top_p=1.0,
 
217
  "content": query},
218
  ]
219
  completion = client.chat.completions.create(
220
+ # model="moonshot-v1-128k",
221
+ model="moonshot-v1-32k",
222
  messages=messages,
223
  temperature=temperature,
224
  top_p=1.0,