From 092e87211e4ed83c090df22bc168b2483739f772 Mon Sep 17 00:00:00 2001
From: Daniel Strobusch <1847260+dastrobu@users.noreply.github.com>
Date: Sat, 23 Dec 2023 16:09:49 +0100
Subject: [PATCH] fix bad convert parameter (#178)

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

diff --git a/llms/mixtral/convert.py b/llms/mixtral/convert.py
index 023bf9a5..6992f5c4 100644
--- a/llms/mixtral/convert.py
+++ b/llms/mixtral/convert.py
@@ -15,7 +15,7 @@ from mixtral import Mixtral, ModelArgs
 from mlx.utils import tree_flatten, tree_map, tree_unflatten
 
 
-def convert(weights, config):
+def convert(tf, config):
     def convert_single(k, v):
         v = v.to(torch.float16).numpy()
         if "block_sparse_moe" not in k: