diff --git a/unsloth/models/gemma.py b/unsloth/models/gemma.py index dcf4b0fa30..b626c79ff3 100644 --- a/unsloth/models/gemma.py +++ b/unsloth/models/gemma.py @@ -113,8 +113,8 @@ class FastGemmaRotaryEmbedding(torch.nn.Module): new_cos = emb.cos().to(dtype=x.dtype)[:,:seq_len] new_sin = emb.sin().to(dtype=x.dtype)[:,:seq_len] - print(new_cos) - print(old_cos) + print(new_cos, new_cos.shape) + print(old_cos, old_cos.shape) raise 1 return new_cos, new_sin pass