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

Merge branch 'dev' into sd3

Browse Source
This commit is contained in:
Kohya S
2026-02-26 08:22:30 +09:00
parent 2217704ce1 1a3ec9ea74
commit d633b51126
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
finetune/tag_images_by_wd14_tagger.py
View File

@@ -404,7 +404,7 @@ def main(args):
rating_tag = None
quality_max_prob = -1
quality_tag = None
character_tags = []
img_character_tags = []
min_thres = min(
args.thresh,
@@ -449,7 +449,7 @@ def main(args):
tag_freq[tag_name] = tag_freq.get(tag_name, 0) + 1
character_tag_text += caption_separator + tag_name
if args.character_tags_first: # we separate character tags
character_tags.append((tag_name, p))
img_character_tags.append((tag_name, p))
else:
combined_tags.append((tag_name, p))
elif (
@@ -464,9 +464,9 @@ def main(args):
# sort by probability
combined_tags.sort(key=lambda x: x[1], reverse=True)
if character_tags:
character_tags.sort(key=lambda x: x[1], reverse=True)
combined_tags = character_tags + combined_tags
if img_character_tags:
img_character_tags.sort(key=lambda x: x[1], reverse=True)
combined_tags = img_character_tags + combined_tags
combined_tags = [t[0] for t in combined_tags] # remove probability
if quality_tag is not None: