From 7c0e4b15de4fa2802589c13ab245b3019af57930 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 25 Jun 2025 02:09:56 -0700 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 0cdd51733b..9db8abdd43 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1130,9 +1130,7 @@ def CausalLM_fast_forward(fast_forward_inference): # Output last hidden states without logits if asked if os.environ.get("UNSLOTH_RETURN_HIDDEN_STATES", "0") == "1": if num_logits_to_keep != 0: - print("llama 1133 hidden_states", hidden_states.shape) hidden_states = hidden_states[:, -num_logits_to_keep:, :] - print("llama 1133 hidden_states", hidden_states.shape) return CausalLMOutputWithPast( loss = None, logits = hidden_states,