[Lint] Fix logits processor formatting on main (#40303)

2026-09-19 14:31:56 +08:00
parent 111aeedd37
commit 6533223502
1 changed files with 1 additions and 3 deletions
@@ -464,9 +464,7 @@ class LogitsProcessor(nn.Module):
        ):
            parallel = get_parallel()
            group = (
-                parallel.attn_tp_group
+                parallel.attn_tp_group if self.use_attn_tp_group else parallel.tp_group
                if self.use_attn_tp_group
                else parallel.tp_group
            )
            chunking_group = group.cpu_group
        self.input_logprob_processor = InputLogprobProcessor(