From 7380ebfb0de1b5a0837aaca0ba3df30bee0742ed Mon Sep 17 00:00:00 2001 From: Anchen Date: Thu, 11 Jan 2024 05:53:52 -0800 Subject: [PATCH] fix: undefined hf_path (#292) --- llms/hf_llm/convert.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llms/hf_llm/convert.py b/llms/hf_llm/convert.py index 4a899ab8..3523c7f2 100644 --- a/llms/hf_llm/convert.py +++ b/llms/hf_llm/convert.py @@ -29,9 +29,9 @@ def fetch_from_hub(model_path: str, local: bool): for wf in weight_files: weights.update(mx.load(wf).items()) - config = transformers.AutoConfig.from_pretrained(hf_path) + config = transformers.AutoConfig.from_pretrained(model_path) tokenizer = transformers.AutoTokenizer.from_pretrained( - hf_path, + model_path, ) return weights, config.to_dict(), tokenizer