Update llama.py

2025-02-12 23:03:40 -08:00 · 2025-02-12 23:03:40 -08:00 · a3af8e3718
commit a3af8e3718
parent 36bf805fa3
1 changed files with 0 additions and 7 deletions
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@ -1162,13 +1162,6 @@ def CausalLM_fast_forward(fast_forward_inference):
            output = (logits,) + outputs[1:]
            return (loss,) + output if loss is not None else output
        print("***", logits.dtype, logits.shape)
-        print(CausalLMOutputWithPast(
-            loss=loss,
-            logits=logits,
-            past_key_values=outputs.past_key_values,
-            hidden_states=outputs.hidden_states,
-            attentions=outputs.attentions,
-        ))
        return CausalLMOutputWithPast(
            loss=loss,
            logits=logits,