diff --git a/vllm/model_executor/models/minicpmv.py b/vllm/model_executor/models/minicpmv.py index 001329b176..add63e169f 100644 --- a/vllm/model_executor/models/minicpmv.py +++ b/vllm/model_executor/models/minicpmv.py @@ -217,7 +217,9 @@ class Resampler2_5(BaseResampler): for i in range(bs): tgt_h, tgt_w = tgt_sizes[i].tolist() pos_embed.append( - self.pos_embed[:tgt_h, :tgt_w, :].reshape((tgt_h * tgt_w, -1)).to(dtype) + self.pos_embed[:tgt_h, :tgt_w, :] + .reshape((tgt_h * tgt_w, -1)) + .to(device=device, dtype=dtype) ) # patches * D key_padding_mask[i, patch_len[i] :] = True pos_embed = torch.nn.utils.rnn.pad_sequence(