Skip to content

Commit 41a355e

Browse files
committed
update topp&topk&temperature init of gradio demo
1 parent df8c40d commit 41a355e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

moss_web_demo_gradio.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def predict(input, chatbot, max_length, top_p, temperature, history):
126126
attention_mask=inputs.attention_mask.cuda(),
127127
max_length=max_length,
128128
do_sample=True,
129-
top_k=50,
129+
top_k=40,
130130
top_p=top_p,
131131
temperature=temperature,
132132
num_return_sequences=1,
@@ -166,10 +166,10 @@ def reset_state():
166166
emptyBtn = gr.Button("Clear History")
167167
max_length = gr.Slider(
168168
0, 4096, value=2048, step=1.0, label="Maximum length", interactive=True)
169-
top_p = gr.Slider(0, 1, value=0.7, step=0.01,
169+
top_p = gr.Slider(0, 1, value=0.8, step=0.01,
170170
label="Top P", interactive=True)
171171
temperature = gr.Slider(
172-
0, 1, value=0.95, step=0.01, label="Temperature", interactive=True)
172+
0, 1, value=0.7, step=0.01, label="Temperature", interactive=True)
173173

174174
history = gr.State([]) # (message, bot_message)
175175

0 commit comments

Comments
 (0)