We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
logical_not
1 parent e47433b commit 2e1b8bcCopy full SHA for 2e1b8bc
vllm/model_executor/models/midashenglm.py
@@ -426,8 +426,7 @@ def forward(
426
assert x_length.ndim == 1, "Lengths are of size (B,)"
427
scaled_lengths = (x_length / (self.hop_length * 4)).long()
428
mask = self._to_mask(max_length=t, lengths=scaled_lengths)
429
- split_masks = mask.logical_not().split(target_length_in_patches,
430
- dim=-1)
+ split_masks = mask.split(target_length_in_patches, dim=-1)
431
else:
432
mask = None
433
split_masks = [None] * len(input_splits)
0 commit comments