diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index e03f73301d..eaf4f8b732 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1153,14 +1153,7 @@ def CausalLM_fast_forward(fast_forward_inference): if not return_dict: output = (logits,) + outputs[1:] return (loss,) + output if loss is not None else output - - print(CausalLMOutputWithPast( - loss=loss, - logits=logits, - past_key_values=outputs.past_key_values, - hidden_states=outputs.hidden_states, - attentions=outputs.attentions, - )) + return CausalLMOutputWithPast( loss=loss, logits=logits,