Spaces:
Running
Running
蒲源
commited on
Commit
·
ae77d05
1
Parent(s):
47391ff
polish(pu): set moonshot-v1-32k as default model
Browse files- app_mqa_database.py +1 -1
- 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=
|
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,
|