From 6604654b2f3a37bea2a1dc314498f498aa1b6dde Mon Sep 17 00:00:00 2001 From: Tong Li Date: Mon, 31 Mar 2025 13:15:18 +0800 Subject: [PATCH] minor 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 bb7b84826..bb719a13c 100644 --- a/applications/ColossalChat/rl_example.py +++ b/applications/ColossalChat/rl_example.py @@ -32,7 +32,7 @@ if __name__ == "__main__": help="Number of prompts per device. Number of samples = tMbs * num of generation per prompt.", ) parser.add_argument("-tmbs", "--train-microbatch-size", type=int, default=2, help="Number of samples per device.") - parser.add_argument("-b", "--backend", type=str, default="transformers", choices=["transformers","vllm"]) + parser.add_argument("-b", "--backend", type=str, default="transformers", choices=["transformers", "vllm"]) parser.add_argument("-a", "--algo", type=str, default="GRPO", choices=["Simple", "GRPO", "EvalGRPO"]) args = parser.parse_args()