From ce708bef1c7a95bf2ba2b244b8c4931dd0ecf62f Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 3 Mar 2025 15:49:04 -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 233f104ecb..c7e630d423 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -857,7 +857,6 @@ def LlamaModel_fast_forward( return module(*inputs, past_key_value, output_attentions, padding_mask = padding_mask, position_embeddings = position_embeddings) return custom_forward pass - layer_outputs = torch.utils.checkpoint.checkpoint( create_custom_forward(decoder_layer), hidden_states,