From a04974168ef53adfa18bcdcc68f9a7a9688fdc33 Mon Sep 17 00:00:00 2001 From: Alex Cheema Date: Sun, 14 Jul 2024 21:46:00 -0700 Subject: [PATCH] fix model import path --- exo/inference/mlx/sharded_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/exo/inference/mlx/sharded_utils.py b/exo/inference/mlx/sharded_utils.py index d4162278..c9a5fb08 100644 --- a/exo/inference/mlx/sharded_utils.py +++ b/exo/inference/mlx/sharded_utils.py @@ -42,7 +42,7 @@ def _get_classes(config: dict): model_type = config["model_type"] model_type = MODEL_REMAPPING.get(model_type, model_type) try: - arch = importlib.import_module(f"inference.mlx.models.{model_type}") + arch = importlib.import_module(f"exo.inference.mlx.models.{model_type}") except ImportError: msg = f"Model type {model_type} not supported." logging.error(msg)