merge grpo-latest

2025-08-31 08:34:14 +00:00 · 2025-04-16 10:03:19 +08:00
parent 447ab74fb4 9467c10690
commit 24c01864df
1 changed files with 2 additions and 1 deletions
--- a/applications/ColossalChat/rl_example.py
+++ b/applications/ColossalChat/rl_example.py
@@ -144,7 +144,8 @@ if __name__ == "__main__":
        grpo_config=grpo_config,
        plugin_config={
            "zero_stage": 2,
-        },  # for zero
+        },
+        # for zero
        # plugin_config={
        #     "pp_size": 2,
        #     "tp_size": 2,