From 3a4681fdd9c640dea7c6fa75d2a393dae3a79823 Mon Sep 17 00:00:00 2001
From: Tong Li <tong.li352711588@gmail.com>
Date: Wed, 11 Jun 2025 17:54:18 +0800
Subject: [PATCH] fix pp memory issue (#6344)

Co-authored-by: Tong Li <tong.li35271158@gmail.com>
---
 applications/ColossalChat/coati/distributed/grpo_consumer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/applications/ColossalChat/coati/distributed/grpo_consumer.py b/applications/ColossalChat/coati/distributed/grpo_consumer.py
index 20b03597a..ec07fe2b5 100644
--- a/applications/ColossalChat/coati/distributed/grpo_consumer.py
+++ b/applications/ColossalChat/coati/distributed/grpo_consumer.py
@@ -358,7 +358,7 @@ class GRPOConsumer(BaseConsumer):
                         criterion=_criterion,
                         optimizer=self.optimizer,
                         return_loss=True,
-                        return_outputs=True,
+                        return_outputs=False,
                     )
                     loss = policy_model_outputs["loss"]