From 6923365bfff4ac92611fcfea28ca199606cec40d Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 12 Feb 2025 19:01:23 -0800 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 7481b833db..1b1da9001f 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -375,6 +375,7 @@ def LlamaAttention_fast_forward( del self.RH_Q del self.attention pass + print(attention_mask) bsz, q_len, _ = hidden_states.size() @@ -449,7 +450,6 @@ def LlamaAttention_fast_forward( else: # Grouped query attention if SDPA_HAS_GQA: - print("=====================") # Needs (batch_size, n_heads, seq_len, head_dim) # is_casual and attention_mask must not be both set! A = scaled_dot_product_attention(Q, K, V, attn_mask = attention_mask, is_causal = False, enable_gqa = n_groups != 1)