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