add huggingface repo url print (#534)

2024-03-06 13:51:31 +08:00 · 2024-03-06 13:51:31 +08:00 · 710c552731
commit 710c552731
parent 5de7c2ac33
1 changed files with 1 additions and 0 deletions
--- a/llms/mlx_lm/utils.py
+++ b/llms/mlx_lm/utils.py
@ -470,6 +470,7 @@ def upload_to_hub(path: str, upload_repo: str, hf_path: str):
        repo_id=upload_repo,
        repo_type="model",
    )
+    print(f"Upload successful, go to https://huggingface.co/{upload_repo} for details.")


 def save_weights(