From 08ccd63882574e6f195de8c044e2a336834fc72e Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 20 Dec 2024 02:47:32 -0800 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 169bfca07a..26b9e07a27 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1055,6 +1055,7 @@ def CausalLM_fast_forward(fast_forward_inference): self.extra_ignored_labels = torch.full((self.max_seq_length, 1), -100, device = "cuda:0") pass + print(kwargs.get("num_items_in_batch", None) or kwargs.get("n_items", None)) shift_labels = torch.hstack((labels[..., 1:], self.extra_ignored_labels[:labels.shape[0]])) loss = fast_cross_entropy_loss( logits = shift_logits,