mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-21 02:15:39 +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)
|
weights = weights.to(dtype=self.dtype)
|
||||||
|
|
||||||
embeds = self.pos_embed(indices)
|
embeds = self.pos_embed(indices)
|
||||||
weighted_embeds = embeds * weights
|
embeds *= weights
|
||||||
combined = weighted_embeds.sum(dim=0)
|
combined = embeds.sum(dim=0)
|
||||||
|
|
||||||
combined = combined.reshape(
|
combined = combined.reshape(
|
||||||
h // m_size, m_size, w // m_size, m_size, hidden_dim
|
h // m_size, m_size, w // m_size, m_size, hidden_dim
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user