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")