forked from Karylab-cklius/vllm
[Bugfix] fix EVS for qwen3-vl (#44205)
Signed-off-by: Rui "Garry" Gao <garrygaogg@gmail.com> Co-authored-by: Cyrus Leung <tlleungac@connect.ust.hk>
This commit is contained in:
co-authored by
Cyrus Leung
parent
90619351e3
commit
4b87b3e845
@@ -2269,6 +2269,8 @@ class Qwen3VLForConditionalGeneration(
|
||||
input_embeds for the LLM.
|
||||
"""
|
||||
|
||||
device = video_embeddings.device
|
||||
|
||||
# Generate video replacement token IDs using get_video_repl
|
||||
# This tokenizes each frame separator independently, then uses pre-tokenized
|
||||
# special tokens to ensure consistent tokenization regardless of
|
||||
@@ -2283,10 +2285,8 @@ class Qwen3VLForConditionalGeneration(
|
||||
select_token_id=self.is_multimodal_pruning_enabled,
|
||||
)
|
||||
|
||||
repl_token_ids = torch.tensor(video_repl.full)
|
||||
embed_token_id = _cached_tensor(
|
||||
self.config.video_token_id, repl_token_ids.device
|
||||
)
|
||||
repl_token_ids = torch.tensor(video_repl.full, device=device)
|
||||
embed_token_id = _cached_tensor(self.config.video_token_id, device=device)
|
||||
is_video_embed = torch.isin(repl_token_ids, embed_token_id)
|
||||
|
||||
# Get text embeddings for indicator tokens (has only `visual_dim``).
|
||||
|
||||
Reference in New Issue
Block a user