change method name, repo is private in default etc

This commit is contained in:
Kohya S
2023-04-05 23:16:49 +09:00
parent 74220bb52c
commit 541539a144
7 changed files with 88 additions and 68 deletions

View File

@@ -30,7 +30,7 @@ def upload(
repo_type = args.huggingface_repo_type
token = args.huggingface_token
path_in_repo = args.huggingface_path_in_repo + dest_suffix
private = args.huggingface_repo_visibility == "private"
private = args.huggingface_repo_visibility is None or args.huggingface_repo_visibility != "public"
api = HfApi(token=token)
if not exists_repo(repo_id=repo_id, repo_type=repo_type, token=token):
api.create_repo(repo_id=repo_id, repo_type=repo_type, private=private)