From 26d9eaddbcb6f71aeb34325cdba41c6494f9b82f Mon Sep 17 00:00:00 2001 From: Tong Li Date: Sun, 27 Apr 2025 17:51:47 +0800 Subject: [PATCH] fix save issue --- applications/ColossalChat/coati/distributed/grpo_consumer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/applications/ColossalChat/coati/distributed/grpo_consumer.py b/applications/ColossalChat/coati/distributed/grpo_consumer.py index d4589b0e2..29307e1d3 100644 --- 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")