diff --git a/applications/ColossalChat/rl_example.py b/applications/ColossalChat/rl_example.py index 1b5c18486..57cab4164 100644 --- a/applications/ColossalChat/rl_example.py +++ b/applications/ColossalChat/rl_example.py @@ -60,7 +60,7 @@ if __name__ == "__main__": ignore_eos=True, include_stop_str_in_output=True, stop=[""], - temperature=0.2, + temperature=0.7, top_p=0.95, ) )