anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[BugFix][Qwen-Image] align txt_seq_lens RoPE width with padded embeds (#4474)

Browse Source 
Signed-off-by: samithuang <285365963@qq.com>
This commit is contained in:
Samit authored and GitHub committed 2026-06-16 19:09:25 +08:00
1 parent 3fa8817fc5
commit 5f5b77aa9c
9 files changed
+151 -17
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vllm_omni/diffusion/worker/input_batch.py
+8 -1
View File
@@ -253,8 +253,15 @@ def _prepare_request_prompt_field(
f"{embeds_attr} for request {state.request_id} requires seq_len "
f"{max_actual_seq_len}, got target {target_seq_len}."
)
return embeds[:, :target_seq_len], None if mask is None else mask[:, :target_seq_len]
embeds = embeds[:, :target_seq_len]
mask = None if mask is None else mask[:, :target_seq_len]
setattr(state, embeds_attr, embeds)
if mask is not None:
setattr(state, mask_attr, mask)
setattr(state, seq_lens_attr, [int(embeds.shape[1])] * int(embeds.shape[0]))
return embeds, mask
setattr(state, seq_lens_attr, [int(embeds.shape[1])] * int(embeds.shape[0]))
return embeds, mask