Update llama.py

2025-03-21 15:44:38 -07:00 · 2025-03-21 15:44:38 -07:00 · 8aeb3c0e1c
commit 8aeb3c0e1c
parent 9985182ed0
1 changed files with 1 additions and 0 deletions
--- 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,