diff --git a/src/setfit/span/trainer.py b/src/setfit/span/trainer.py index 9c417c3a..2187b06a 100644 --- a/src/setfit/span/trainer.py +++ b/src/setfit/span/trainer.py @@ -291,8 +291,8 @@ def push_to_hub(self, repo_id: str, polarity_repo_id: Optional[str] = None, **kw Configuration object to be saved alongside the model weights. commit_message (`str`, *optional*): Message to commit while pushing. - private (`bool`, *optional*, defaults to `False`): - Whether the repository created should be private. + private (`bool`, *optional*): + Whether to make the repo private. If `None` (default), the repo will be public unless the organization's default is private. This value is ignored if the repo already exists. api_endpoint (`str`, *optional*): The API endpoint to use when pushing the model to the hub. token (`str`, *optional*): diff --git a/src/setfit/trainer.py b/src/setfit/trainer.py index 20411d3b..7651d55c 100644 --- a/src/setfit/trainer.py +++ b/src/setfit/trainer.py @@ -792,8 +792,8 @@ def push_to_hub(self, repo_id: str, **kwargs) -> str: Configuration object to be saved alongside the model weights. commit_message (`str`, *optional*): Message to commit while pushing. - private (`bool`, *optional*, defaults to `False`): - Whether the repository created should be private. + private (`bool`, *optional*): + Whether to make the repo private. If `None` (default), the repo will be public unless the organization's default is private. This value is ignored if the repo already exists. api_endpoint (`str`, *optional*): The API endpoint to use when pushing the model to the hub. token (`str`, *optional*):