From 041aa7d909c040028b24f76ad4c911ff7f8163b3 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sat, 24 Feb 2024 17:18:09 +1100 Subject: [PATCH] Update gemma.py --- unsloth/models/gemma.py | 1 - 1 file changed, 1 deletion(-) diff --git a/unsloth/models/gemma.py b/unsloth/models/gemma.py index d9172d2c3b..176d544533 100644 --- a/unsloth/models/gemma.py +++ b/unsloth/models/gemma.py @@ -67,7 +67,6 @@ def GemmaDecoderLayer_fast_forward( # padding_mask: Optional[torch.LongTensor] = None, *args, **kwargs, ) -> Tuple[torch.FloatTensor, Optional[Tuple[torch.FloatTensor, torch.FloatTensor]]]: - print(past_key_value) if False:#past_key_value is not None: do_prefill = not hasattr(self.self_attn, "paged_attention")