From 4c5ebcc960393131841cc3685b9155722ceca2bd Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 28 Jan 2024 03:56:54 +1100 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 9ce868150f..4fc4071968 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -635,7 +635,7 @@ def LlamaForCausalLM_fast_forward( logits = self.lm_head(hidden_states) loss = None - print(labels) + print("labels", labels) if labels is not None: shift_logits = logits if not hasattr(self, "extra_ignored_labels"):