From da867a4d8f9db053422accf1d81b0fc8dc55069c Mon Sep 17 00:00:00 2001 From: YeAnbang Date: Sat, 3 May 2025 10:08:12 +0800 Subject: [PATCH] small fix --- applications/ColossalChat/rl_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/applications/ColossalChat/rl_example.py b/applications/ColossalChat/rl_example.py index 817843d59..6611835f2 100644 --- a/applications/ColossalChat/rl_example.py +++ b/applications/ColossalChat/rl_example.py @@ -184,7 +184,7 @@ if __name__ == "__main__": enforce_eager=True, enable_chunked_prefill=True, max_model_len=args.max_new_tokens + args.max_prompt_tokens, - tensor_parallel_size=2, + tensor_parallel_size=1, ) ) generate_config.update(