Skip to content

Commit

Permalink
add persistence to max_tokens temperature and top_p
Browse files Browse the repository at this point in the history
Signed-off-by: Mingxin Zheng <[email protected]>
  • Loading branch information
mingxin-zheng committed Nov 4, 2024
1 parent a959c8c commit 02b2c69
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion m3/demo/gradio_m3.py
Original file line number Diff line number Diff line change
Expand Up @@ -422,7 +422,7 @@ def squash_expert_messages_into_user(self, messages: list):
j += 1
del messages[i + 1 : j] # Remove all the squashed expert messages

i += 1 # Move to the next message. TODO: Check if this is correct
i += 1

return messages

Expand Down Expand Up @@ -530,6 +530,9 @@ def process_prompt(self, prompt, sv, chat_history):
use_model_cards=sv.use_model_cards,
download_file_path=download_pkg,
temp_working_dir=sv.temp_working_dir,
max_tokens=sv.max_tokens,
temperature=sv.temperature,
top_p=sv.top_p,
interactive=True,
sys_msgs_to_hide=sv.sys_msgs_to_hide,
image_url_backup=sv.image_url,
Expand Down

0 comments on commit 02b2c69

Please sign in to comment.