Spaces:
Runtime error
Runtime error
蒲源
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,
|