incorrect condition on self.sliding_window_size (#2289)

laxmareddyp · web-flow · commit cf8aedfffd01 · 2025-06-17T17:36:07.000-07:00
diff --git a/keras_hub/src/models/qwen3/qwen3_attention.py b/keras_hub/src/models/qwen3/qwen3_attention.py
@@ -299,7 +299,7 @@ def _compute_attention(
             attention_scores,
             ops.cast(self._inv_norm_factor, self.compute_dtype),
         )
-        if not self.sliding_window_size:
+        if self.sliding_window_size:
             attention_mask = self._mask_sliding_window(
                 attention_mask,
                 cache_update_index=cache_update_index

Original file line number	Diff line number	Diff line change
`@@ -299,7 +299,7 @@ def _compute_attention(`
`299`	`299`	`attention_scores,`
`300`	`300`	`ops.cast(self._inv_norm_factor, self.compute_dtype),`
`301`	`301`	`)`
`302`		`- if not self.sliding_window_size:`
	`302`	`+ if self.sliding_window_size:`
`303`	`303`	`attention_mask = self._mask_sliding_window(`
`304`	`304`	`attention_mask,`
`305`	`305`	`cache_update_index=cache_update_index`