zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 04:14:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix upload to hub for HF LLMs conversion script. (#221)

Browse Source 
* Fix upload to hub snippet.

* Weights -> model.

* reverting last commit.
This commit is contained in:
Vaibhav Srivastav
2024-01-04 19:36:05 +05:30
committed by GitHub
parent fc9dde34c9
commit a9ee883bf7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
llms/hf_llm/convert.py
View File

@@ -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 @@ if __name__ == "__main__":
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)