replace group

InternLM · Jan 24, 2025 · 713aad5 · 713aad5
1 parent ada2cc3
commit 713aad5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/pytorch/engine/engine.py b/lmdeploy/pytorch/engine/engine.py
@@ -1030,7 +1030,7 @@ def _finally_rank0(self):
         else:
             # broadcast exit signal.
             try:
-                broadcast_inputs(0, [None, None, None], group=self.dist_ctx.world_cpu_group)
+                broadcast_inputs(0, [None, None, None], group=self.dist_ctx.tp_cpu_group)
             except Exception:
                 logger.warning('Can not broadcast exit signal.')