[Model][Qwen3VL] Slighly speedup fast_pos_embed_interpolate (#28434)

Signed-off-by: Lukas Geiger <lukas.geiger94@gmail.com>
This commit is contained in:
Lukas Geiger 2025-11-11 10:35:10 +00:00 committed by GitHub
parent c7991269dd
commit 9973e6e04a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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