mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-20 19:55:01 +08:00
[Model][Qwen3VL] Slighly speedup fast_pos_embed_interpolate (#28434)
Signed-off-by: Lukas Geiger <lukas.geiger94@gmail.com>
This commit is contained in:
parent
c7991269dd
commit
9973e6e04a
@ -491,8 +491,8 @@ class Qwen3_VisionTransformer(nn.Module):
|
||||
weights = weights.to(dtype=self.dtype)
|
||||
|
||||
embeds = self.pos_embed(indices)
|
||||
weighted_embeds = embeds * weights
|
||||
combined = weighted_embeds.sum(dim=0)
|
||||
embeds *= weights
|
||||
combined = embeds.sum(dim=0)
|
||||
|
||||
combined = combined.reshape(
|
||||
h // m_size, m_size, w // m_size, m_size, hidden_dim
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user