Skip to content

Commit 2e1b8bc

Browse files
authored
[Model][Bugfix] Fix MiDashengLM audio encoder mask by removing incorrect logical_not (vllm-project#25925)
Signed-off-by: zhoukz <[email protected]>
1 parent e47433b commit 2e1b8bc

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

vllm/model_executor/models/midashenglm.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -426,8 +426,7 @@ def forward(
426426
assert x_length.ndim == 1, "Lengths are of size (B,)"
427427
scaled_lengths = (x_length / (self.hop_length * 4)).long()
428428
mask = self._to_mask(max_length=t, lengths=scaled_lengths)
429-
split_masks = mask.logical_not().split(target_length_in_patches,
430-
dim=-1)
429+
split_masks = mask.split(target_length_in_patches, dim=-1)
431430
else:
432431
mask = None
433432
split_masks = [None] * len(input_splits)

0 commit comments

Comments
 (0)