diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index ad1b6f4943..4feee54f35 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -892,6 +892,7 @@ def LlamaModel_fast_forward( if output_hidden_states: all_hidden_states += (hidden_states,) next_cache = next_decoder_cache if use_cache else None + print(next_cache) if not return_dict: return tuple(v for v in [hidden_states, next_cache, all_hidden_states, all_self_attns] if v is not None)