From cc4c71d8cfdaeacd5386c985bf540ca8bba8cb85 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 3 Mar 2025 15:48:55 -0800 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 7070919903..233f104ecb 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -857,8 +857,7 @@ def LlamaModel_fast_forward( return module(*inputs, past_key_value, output_attentions, padding_mask = padding_mask, position_embeddings = position_embeddings) return custom_forward pass - - print(torch.utils.checkpoint.checkpoint) + layer_outputs = torch.utils.checkpoint.checkpoint( create_custom_forward(decoder_layer), hidden_states,