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,