sampling, printing xD
Browse files
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=
|
| 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__":
|