diff --git a/vllm_omni/diffusion/models/minimax_h3/chunked_decode.py b/vllm_omni/diffusion/models/minimax_h3/chunked_decode.py index 88cf5395a..213cb4405 100644 --- a/vllm_omni/diffusion/models/minimax_h3/chunked_decode.py +++ b/vllm_omni/diffusion/models/minimax_h3/chunked_decode.py @@ -232,10 +232,7 @@ class _MiniMaxH3ChunkDecodeCoordinator: dtype=torch.int64, device=device, ) - gathered_signatures = [ - torch.empty_like(padded_signature) - for _ in range(dist.get_world_size(group)) - ] + gathered_signatures = [torch.empty_like(padded_signature) for _ in range(dist.get_world_size(group))] dist.all_gather(gathered_signatures, padded_signature, group=group) if any(not torch.equal(padded_signature, peer) for peer in gathered_signatures): raise MiniMaxH3ChunkedDecodeUnsupportedError(