From 47cbc418e20fea05603e7c021d68df592e115afb Mon Sep 17 00:00:00 2001
From: Alex Barron <abarron22@apple.com>
Date: Fri, 6 Dec 2024 16:33:25 -0800
Subject: [PATCH] relative import

---
 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 b620b662..f74a7757 100644
--- a/llms/mlx_lm/evaluate.py
+++ b/llms/mlx_lm/evaluate.py
@@ -14,9 +14,9 @@ import mlx.nn as nn
 import numpy as np
 from lm_eval.api.model import LM
 from lm_eval.api.registry import register_model
-from mlx_lm.models.cache import make_prompt_cache
 from tqdm import tqdm
 
+from .models.cache import make_prompt_cache
 from .utils import load, stream_generate
 
 PAD = 0