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

This commit is contained in:
Xiaoyu Zhang
2026-09-19 14:31:56 +08:00
committed by GitHub
parent 111aeedd37
commit 6533223502
+1 -3
View File
@@ -464,9 +464,7 @@ class LogitsProcessor(nn.Module):
):
parallel = get_parallel()
group = (
parallel.attn_tp_group
if self.use_attn_tp_group
else parallel.tp_group
parallel.attn_tp_group if self.use_attn_tp_group else parallel.tp_group
)
chunking_group = group.cpu_group
self.input_logprob_processor = InputLogprobProcessor(