From dd1690df81931c8472fb7fa028ff897a9efb1329 Mon Sep 17 00:00:00 2001 From: Sindhu Satish Date: Wed, 29 Jan 2025 06:00:06 -0800 Subject: [PATCH] bug fix --- llms/mlx_lm/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llms/mlx_lm/utils.py b/llms/mlx_lm/utils.py index 03d22626..8e48ab25 100644 --- a/llms/mlx_lm/utils.py +++ b/llms/mlx_lm/utils.py @@ -709,7 +709,7 @@ def load( model_config={}, adapter_path: Optional[str] = None, lazy: bool = False, - commit_hash: Optional[str] = "main", + revision: Optional[str] = "main", ) -> Tuple[nn.Module, TokenizerWrapper]: """ Load the model and tokenizer from a given path or a huggingface repository.