From 8cb12078f9536b7936817e9b068a6ddd7b4686dd Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sat, 24 Feb 2024 17:15:44 +1100 Subject: [PATCH] Update gemma.py --- unsloth/models/gemma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/gemma.py b/unsloth/models/gemma.py index a8825352b5..69c4535d44 100644 --- a/unsloth/models/gemma.py +++ b/unsloth/models/gemma.py @@ -93,7 +93,7 @@ def GemmaDecoderLayer_fast_forward( hidden_states = fast_rms_layernorm(self.input_layernorm, hidden_states) hidden_states, self_attn_weights, present_key_value = self.self_attn( hidden_states=hidden_states, - causal_mask=causal_mask, + # causal_mask=causal_mask, attention_mask=attention_mask, position_ids=position_ids, past_key_value=past_key_value,