diff --git a/llms/hf_llm/convert.py b/llms/hf_llm/convert.py index 2bc48fe23..0704b24bc 100644 --- a/llms/hf_llm/convert.py +++ b/llms/hf_llm/convert.py @@ -69,7 +69,7 @@ def make_shards(weights: dict, max_file_size_gibibyte: int = 15): return shards -def upload_to_hub(path: str, name: str): +def upload_to_hub(path: str, name: str, hf_path: str): import os from huggingface_hub import HfApi, ModelCard, logging @@ -171,4 +171,4 @@ def upload_to_hub(path: str, name: str): json.dump(config, fid, indent=4) if args.upload_name is not None: - upload_to_hub(mlx_path, args.upload_name) + upload_to_hub(mlx_path, args.upload_name, args.hf_path)