[Bugfix][MiniMax-H3] Apply pre-commit formatting
Signed-off-by: hsliuustc0106 <liuhongsheng4@huawei.com>
This commit is contained in:
1 parent
f9c9de8d16
commit
81de3f3308
1 file changed
+1
-4
@@ -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(
|
||||
|
||||
Reference in new issue
Block a user