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

fix: update hf_hub_download parameters to fix wd14 tagger regression

Browse Source
This commit is contained in:
Glen
2025-04-19 11:54:12 -06:00
parent 52c8dec953
commit 26db64be17
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
finetune/tag_images_by_wd14_tagger.py
View File

@@ -100,15 +100,19 @@ def main(args):
else: else:
for file in SUB_DIR_FILES: for file in SUB_DIR_FILES:
hf_hub_download( hf_hub_download(
args.repo_id, repo_id=args.repo_id,
file, filename=file,
subfolder=SUB_DIR, subfolder=SUB_DIR,
cache_dir=os.path.join(model_location, SUB_DIR), local_dir=os.path.join(model_location, SUB_DIR),
force_download=True, force_download=True,
force_filename=file,
) )
for file in files: for file in files:
hf_hub_download(args.repo_id, file, cache_dir=model_location, force_download=True, force_filename=file) hf_hub_download(
repo_id=args.repo_id,
filename=file,
local_dir=model_location,
force_download=True,
)
else: else:
logger.info("using existing wd14 tagger model") logger.info("using existing wd14 tagger model")