fix save issue (#6279)

Co-authored-by: Tong Li <tong.li35271158@gmail.com>
2025-08-19 08:27:23 +00:00 · 2025-04-27 17:54:06 +08:00 · 2025-04-27 17:54:06 +08:00 · 37a8be7651
commit 37a8be7651
parent 673682e716
1 changed files with 1 additions and 0 deletions
--- a/applications/ColossalChat/coati/distributed/grpo_consumer.py
+++ b/applications/ColossalChat/coati/distributed/grpo_consumer.py
@ -62,6 +62,7 @@ class GRPOConsumer(BaseConsumer):
            model_config,
            plugin_config,
            minibatch_size,
+            save_interval=save_interval,
            save_dir=save_dir,
        )
        path = model_config.pop("path")