From 318a59f6b4a097fdc329f6b26bd5ab388a4d1b8b Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 12 Feb 2025 20:29:10 -0800 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 1 - 1 file changed, 1 deletion(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index fbc6d53afb..653ebb351f 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -706,7 +706,6 @@ def LlamaModel_fast_forward( pass # Ignore attention_mask - print(attention_mask, attention_mask.shape, attention_mask.dtype) if attention_mask is None: padding_mask = None elif attention_mask is None and self.training: