Spaces:

wdndev
/

tiny_llm_sft_92m_demo

Sleeping

wdndev commited on May 1, 2024

Commit

f0f7502

1 Parent(s): 574a1d4

app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ max_new_tokens = st.sidebar.slider("max_new_tokens", 0, 1024, 512, step=1)
 top_p = st.sidebar.slider("top_p", 0.0, 1.0, 0.8, step=0.01)
 top_k = st.sidebar.slider("top_k", 0, 100, 0, step=1)
 temperature = st.sidebar.slider("temperature", 0.0, 2.0, 1.0, step=0.01)
-do_sample = st.sidebar.checkbox("do_sample", value=True)
 def main():
     model, tokenizer, generation_config = load_model_tokenizer()

 top_p = st.sidebar.slider("top_p", 0.0, 1.0, 0.8, step=0.01)
 top_k = st.sidebar.slider("top_k", 0, 100, 0, step=1)
 temperature = st.sidebar.slider("temperature", 0.0, 2.0, 1.0, step=0.01)
+do_sample = st.sidebar.checkbox("do_sample", value=False)
 def main():
     model, tokenizer, generation_config = load_model_tokenizer()