Update llama.py

2024-11-25 21:52:42 -08:00 · 2024-11-25 21:52:42 -08:00 · f7edb152e7
commit f7edb152e7
parent b93e9cda3b
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@ -1002,6 +1002,7 @@ def CausalLM_fast_forward(fast_forward_inference):
                    attentions=outputs.attentions,
                )
                print(output)
+                print(output)
                return output
            pass
            logits = self.lm_head(hidden_states.to(lm_head.dtype))