diff --git a/sentence_transformers/SentenceTransformer.py b/sentence_transformers/SentenceTransformer.py index 5fbae8177..b9f08df7f 100644 --- a/sentence_transformers/SentenceTransformer.py +++ b/sentence_transformers/SentenceTransformer.py @@ -465,8 +465,8 @@ def save_to_hub(self, endpoint = "https://huggingface.co" repo_url = HfApi(endpoint=endpoint).create_repo( - token, - repo_name, + token=token, + repo_id=repo_name, organization=organization, private=private, repo_type=None,