From c3e51fcf021dd7ce40a7f0add526dcb985dbd6df Mon Sep 17 00:00:00 2001
From: Alex Barron <abarron22@apple.com>
Date: Fri, 6 Dec 2024 16:26:26 -0800
Subject: [PATCH] typo

---
 llms/mlx_lm/evaluate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llms/mlx_lm/evaluate.py b/llms/mlx_lm/evaluate.py
index ff49a1bd..95cf0f1b 100644
--- a/llms/mlx_lm/evaluate.py
+++ b/llms/mlx_lm/evaluate.py
@@ -101,7 +101,7 @@ class MLXLM(LM):
                 targets[:, i : i + step_size] == mx.argmax(logits, axis=-1)
             )
 
-            mx.eval(score, is_greedy)
+            mx.eval(score, ig)
             mx.metal.clear_cache()
 
             is_greedy.append(ig)