From 1a97169fab17ee35136f525b6c924d936a7ff03b Mon Sep 17 00:00:00 2001 From: YWMditto <862779238@qq.com> Date: Tue, 7 Nov 2023 23:07:27 +0800 Subject: [PATCH] update web_demo.py --- web_demo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/web_demo.py b/web_demo.py index 95e2cf8..86c9b2f 100644 --- a/web_demo.py +++ b/web_demo.py @@ -36,9 +36,9 @@ def load_model(): def prepare_generation_config(): with st.sidebar: - max_length = st.slider("Max Length", min_value=32, max_value=2048, value=2048) + max_length = st.slider("Max Length", min_value=32, max_value=16000, value=8000) top_p = st.slider("Top P", 0.0, 1.0, 0.8, step=0.01) - temperature = st.slider("Temperature", 0.0, 1.0, 0.7, step=0.01) + temperature = st.slider("Temperature", 0.0, 1.0, 0.8, step=0.01) st.button("Clear Chat History", on_click=on_btn_click) generation_config = GenerationConfig(max_length=max_length, top_p=top_p, temperature=temperature)