From 66a4380d084f67e8c207b596ae9caffcf15e540e Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 25 Feb 2024 01:50:42 +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 f542b8ef53..8f173e210f 100644 --- a/unsloth/models/gemma.py +++ b/unsloth/models/gemma.py @@ -98,7 +98,6 @@ def GemmaAttention_fast_forward( Q, K = fast_rope_embedding(Q, K, cos, sin) else: cos, sin = self.rotary_emb(V, position_ids, seq_len=None) - print(cos.shape, sin.shape) Q, K = inplace_rope_embedding(Q, K, cos, sin, position_ids) pass