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

[diffusion] fix: remove boolean arithmetic guard to fix compiling (#27065)

Browse Source
This commit is contained in:
Aleksi Vesanto
2026-06-10 13:55:47 +08:00
committed by GitHub
parent 4704b10d0d
commit 08ceb96ea5
1 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/layers/attention/layer.py
+3 -9
View File
@@ -648,15 +648,9 @@ class USPAttention(nn.Module):
sp_size = get_ulysses_parallel_world_size() sp_size = get_ulysses_parallel_world_size()
if ( if (
sum( (num_replicated_prefix > 0 and num_replicated_suffix > 0)
bool(n) or (num_replicated_prefix > 0 and num_replicated_kv_prefix > 0)
for n in ( or (num_replicated_suffix > 0 and num_replicated_kv_prefix > 0)
num_replicated_prefix,
num_replicated_suffix,
num_replicated_kv_prefix,
)
)
> 1
): ):
raise ValueError( raise ValueError(
"USPAttention supports at most one replicated-token mode per call." "USPAttention supports at most one replicated-token mode per call."