diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py
index 6f954bf9a2..f6c337ae36 100644
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@@ -1018,8 +1018,9 @@ def CausalLM_fast_forward(fast_forward_inference):
         *args, **kwargs,
     ) -> Union[Tuple, CausalLMOutputWithPast]:
         # Check for uninitialized DynamicCache
-        print(past_key_values, len(past_key_values))
-        if past_key_values is not None and len(past_key_values) != 0:
+        if past_key_values is not None and len(past_key_values) == 0:
+            past_key_values = None
+        if past_key_values is not None:
             outputs = fast_forward_inference(
                 self,
                 input_ids,