From 07d89f2264f2faa24332f4c4070163f1ec3e45d7 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 12 Feb 2025 03:08:40 -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 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,