minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[CI] Fix lint brought by #27527 (#28988)

Browse Source 
Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
This commit is contained in:
Yuan Luo
2026-06-22 20:40:06 -07:00
committed by GitHub
co-authored by luoyuan.luo
parent 6cd8d2869b
commit abb0717174
2 changed files with 68 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/models/deepseek_ocr.py
+3 -3
View File
@@ -1297,7 +1297,7 @@ class CustomQwen2Decoder(nn.Module):
min_dtype = torch.finfo(dtype).min
is_image = token_type_ids == 0 # [B, S]
is_text = token_type_ids == 1 # [B, S]
is_text = token_type_ids == 1 # [B, S]
mask = torch.full(
(batch_size, sequence_length, sequence_length),
@@ -1312,8 +1312,8 @@ class CustomQwen2Decoder(nn.Module):
causal = idx.unsqueeze(0) <= idx.unsqueeze(1) # [S, S]
text_causal = (
is_text.unsqueeze(2) # [B, S, 1]
& is_text.unsqueeze(1) # [B, 1, S]
is_text.unsqueeze(2) # [B, S, 1]
& is_text.unsqueeze(1) # [B, 1, S]
& causal.unsqueeze(0) # [1, S, S]
) # [B, S, S]