github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update web_demo.py

pull/478/head
YWMditto 2023-11-07 23:07:27 +08:00
parent a3946235b2
commit 1a97169fab
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
web_demo.py
View File

@ -36,9 +36,9 @@ def load_model():
def prepare_generation_config(): def prepare_generation_config():
with st.sidebar: 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) 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) st.button("Clear Chat History", on_click=on_btn_click)
generation_config = GenerationConfig(max_length=max_length, top_p=top_p, temperature=temperature) generation_config = GenerationConfig(max_length=max_length, top_p=top_p, temperature=temperature)