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

fix: add return type annotation for process_tag_replacement function and ensure tags are returned

Browse Source
This commit is contained in:
Kohya S
2025-09-03 22:13:40 +09:00
parent 3f01189bd9
commit b1b9dec980
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
finetune/tag_images_by_wd14_tagger.py
View File

@@ -239,7 +239,7 @@ def main(args):
def remove_underscore(tags):
return [tag.replace("_", " ") if len(tag) > 3 else tag for tag in tags]
def process_tag_replacement(tags: list[str], tag_replacements_arg: str):
def process_tag_replacement(tags: list[str], tag_replacements_arg: str) -> list[str]:
# escape , and ; in tag_replacement: wd14 tag names may contain , and ;,
# so user must be specified them like `aa\,bb,AA\,BB;cc\;dd,CC\;DD` which means
# `aa,bb` is replaced with `AA,BB` and `cc;dd` is replaced with `CC;DD`
@@ -258,6 +258,8 @@ def main(args):
if source in tags:
tags[tags.index(source)] = target
return tags
if default_format:
with open(os.path.join(model_location, CSV_FILE), "r", encoding="utf-8") as f:
reader = csv.reader(f)