add huggingface repo url print (#534)

This commit is contained in:
Madroid Ma 2024-03-06 13:51:31 +08:00 committed by GitHub
parent 5de7c2ac33
commit 710c552731
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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(