Spaces:
Paused
Paused
Robin Genolet
commited on
Commit
·
c011bf3
1
Parent(s):
5fd44e9
fix: st init
Browse files
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=
|
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
|