VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

small fix

Browse Source
This commit is contained in:
ddPn08
2023-03-28 08:59:23 +09:00
parent 054fb3308c
commit 45381b188c
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
library/utils.py
View File

@@ -14,12 +14,12 @@ def fire_in_thread(f):
def huggingface_exists_repo(
repo_id: str, repo_type: str, revision: str = "main", hf_token: str = None
repo_id: str, repo_type: str, revision: str = "main", token: str = None
):
api = HfApi()
try:
api.repo_info(
repo_id=repo_id, token=hf_token, revision=revision, repo_type=repo_type
repo_id=repo_id, token=token, revision=revision, repo_type=repo_type
)
return True
except:
@@ -34,15 +34,15 @@ def huggingface_upload(
):
repo_id = args.huggingface_repo_id
repo_type = args.huggingface_repo_type
hf_token = args.huggingface_token
token = args.huggingface_token
path_in_repo = args.huggingface_path_in_repo + dest_suffix
private = args.huggingface_repo_visibility == "private"
api = HfApi()
if not huggingface_exists_repo(
repo_id=repo_id, repo_type=repo_type, token=hf_token
repo_id=repo_id, repo_type=repo_type, token=token
):
api.create_repo(
token=hf_token, repo_id=repo_id, repo_type=repo_type, private=private
token=token, repo_id=repo_id, repo_type=repo_type, private=private
)
is_folder = (type(src) == str and os.path.isdir(src)) or (
@@ -54,7 +54,7 @@ def huggingface_upload(
repo_type=repo_type,
folder_path=src,
path_in_repo=path_in_repo,
token=hf_token,
token=token,
)
else:
api.upload_file(
@@ -62,5 +62,5 @@ def huggingface_upload(
repo_type=repo_type,
path_or_fileobj=src,
path_in_repo=path_in_repo,
token=hf_token,
token=token,
)