From 5edad55b5d8d17e4ef493d06984a55b337139c1a Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Wed, 31 Jan 2024 19:44:19 +1100 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 6029005265..9663ffce36 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -574,6 +574,7 @@ def LlamaModel_fast_forward( pass # Ignore attention_mask + print(attention_mask) if attention_mask is None: padding_mask = None elif False: