Robin Genolet commited on
Commit
c011bf3
·
1 Parent(s): 5fd44e9

fix: st init

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def display_streamlit_sidebar():
36
  do_sample = form.checkbox('do_sample', value=st.session_state["do_sample"])
37
  top_p = form.slider(label="top_p", min_value=0.0, max_value=1.0, step=0.01, value=st.session_state["top_p"])
38
  top_k = form.slider(label="top_k", min_value=1, max_value=1000, step=1, value=st.session_state["top_k"])
39
- max_new_tokens = form.slider(label="max_new_tokens", min_value=32, max_value=512, step=1, value=st.session_state["max_new_tokens"])
40
  repetition_penalty = form.slider(label="repetition_penalty", min_value=0.0, max_value=5.0, step=0.01, value=st.session_state["repetition_penalty"])
41
 
42
  submitted = form.form_submit_button("Start session")
@@ -70,6 +70,8 @@ def init_session_state():
70
  st.session_state['session_started'] = False
71
  st.session_state["session_events"] = []
72
  st.session_state["model_name_or_path"] = "TheBloke/meditron-7B-GPTQ"
 
 
73
  st.session_state["temperature"] = 0.01
74
  st.session_state["do_sample"] = True
75
  st.session_state["top_p"] = 0.95
 
36
  do_sample = form.checkbox('do_sample', value=st.session_state["do_sample"])
37
  top_p = form.slider(label="top_p", min_value=0.0, max_value=1.0, step=0.01, value=st.session_state["top_p"])
38
  top_k = form.slider(label="top_k", min_value=1, max_value=1000, step=1, value=st.session_state["top_k"])
39
+ max_new_tokens = form.slider(label="max_new_tokens", min_value=32, max_value=16384, step=1, value=st.session_state["max_new_tokens"])
40
  repetition_penalty = form.slider(label="repetition_penalty", min_value=0.0, max_value=5.0, step=0.01, value=st.session_state["repetition_penalty"])
41
 
42
  submitted = form.form_submit_button("Start session")
 
70
  st.session_state['session_started'] = False
71
  st.session_state["session_events"] = []
72
  st.session_state["model_name_or_path"] = "TheBloke/meditron-7B-GPTQ"
73
+ st.session_state["model_name_or_path_other"] = ""
74
+ st.session_state["model_index"] = 0
75
  st.session_state["temperature"] = 0.01
76
  st.session_state["do_sample"] = True
77
  st.session_state["top_p"] = 0.95