diff --git a/llms/mlx_lm/utils.py b/llms/mlx_lm/utils.py index 40102e90..a19b833d 100644 --- a/llms/mlx_lm/utils.py +++ b/llms/mlx_lm/utils.py @@ -169,7 +169,7 @@ def get_model_path(path_or_hf_repo: str, revision: Optional[str] = None) -> Path try: model_path = Path( snapshot_download( - model_id=path_or_hf_repo, + path_or_hf_repo, revision=revision, allow_patterns=[ "*.json",