From 8aeb3c0e1c2d7b5933b75ccdda755ca617899217 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 21 Mar 2025 15:44:38 -0700 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 9f685937c3..6f954bf9a2 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1018,6 +1018,7 @@ 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: outputs = fast_forward_inference( self,