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

2026-06-10 13:55:47 +08:00
parent 4704b10d0d
commit 08ceb96ea5
1 changed files with 3 additions and 9 deletions
@@ -648,15 +648,9 @@ class USPAttention(nn.Module):

        sp_size = get_ulysses_parallel_world_size()
        if (
-            sum(
-                bool(n)
-                for n in (
-                    num_replicated_prefix,
-                    num_replicated_suffix,
-                    num_replicated_kv_prefix,
-                )
-            )
-            > 1
+            (num_replicated_prefix > 0 and num_replicated_suffix > 0)
+            or (num_replicated_prefix > 0 and num_replicated_kv_prefix > 0)
+            or (num_replicated_suffix > 0 and num_replicated_kv_prefix > 0)
        ):
            raise ValueError(
                "USPAttention supports at most one replicated-token mode per call."