diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 4f77280ad7..eaf4f8b732 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1153,8 +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("========== dtype = ", logits.dtype) + return CausalLMOutputWithPast( loss=loss, logits=logits,