Update attention_sink.py

unslothai · danielhanchen · Nov 6, 2025 · Sep 24, 2025 · Sep 24, 2025 · Sep 24, 2025
commit 17a642769b6fbf78339e0015a8228acd1e6f2c1c
diff --git a/unsloth_zoo/flex_attention/attention_sink.py b/unsloth_zoo/flex_attention/attention_sink.py
@@ -182,8 +182,8 @@ def flex_attention_with_sink(
     mask_mod = None
     block_mask = None
     has_flex_cache = hasattr(self_attn, "_flex_attention_cache")
-    has_flex_cache = False
-    is_training = True
+    # has_flex_cache = False
+    # is_training = True
     # Handle inference and training
     if has_static_cache:
         if is_training or (