AccessAndrei commited on
Commit
fcf71d0
·
verified ·
1 Parent(s): 2f2eda6

sampling, printing xD

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -87,7 +87,7 @@ def generate_tt_reply_stream(messages: List[Dict[str, str]]) -> Iterator[str]:
87
  **enc,
88
  streamer=streamer,
89
  max_new_tokens=MAX_NEW_TOKENS,
90
- do_sample=True,
91
  temperature=TEMPERATURE,
92
  top_p=TOP_P,
93
  repetition_penalty=REPETITION_PENALTY,
@@ -147,6 +147,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
147
  return [], [{"role": "system", "content": SYS_PROMPT_TT}]
148
 
149
  clear.click(_reset, inputs=None, outputs=[chatbot, messages_state], queue=False)
 
150
  clear.click(lambda: "", None, msg, queue=False)
151
 
152
  if __name__ == "__main__":
 
87
  **enc,
88
  streamer=streamer,
89
  max_new_tokens=MAX_NEW_TOKENS,
90
+ do_sample=False,
91
  temperature=TEMPERATURE,
92
  top_p=TOP_P,
93
  repetition_penalty=REPETITION_PENALTY,
 
147
  return [], [{"role": "system", "content": SYS_PROMPT_TT}]
148
 
149
  clear.click(_reset, inputs=None, outputs=[chatbot, messages_state], queue=False)
150
+ print("Messages_state: " + messages_state)
151
  clear.click(lambda: "", None, msg, queue=False)
152
 
153
  if __name__ == "__main__":