From e569d8146f4cb94eb5a06c1e4b2d998f3f45af30 Mon Sep 17 00:00:00 2001
From: Divyashree Sreepathihalli <divyashreepathihalli@gmail.com>
Date: Fri, 21 Feb 2025 13:59:49 -0800
Subject: [PATCH] Update gemma_attention.py

---
 keras_hub/src/models/gemma/gemma_attention.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/keras_hub/src/models/gemma/gemma_attention.py b/keras_hub/src/models/gemma/gemma_attention.py
index 22b228f958..680685dd64 100644
--- a/keras_hub/src/models/gemma/gemma_attention.py
+++ b/keras_hub/src/models/gemma/gemma_attention.py
@@ -141,11 +141,8 @@ def _compute_attention(
                 query=q,
                 key=k,
                 value=v,
-                bias=None,
                 mask=attention_mask,
                 scale=query_normalization,
-                is_causal=True,
-                flash_attention=True,
             )
             return attention_output