From e1aedceffa87059dcac0e816132a62fcbd3c123b Mon Sep 17 00:00:00 2001 From: Kohya S <52813779+kohya-ss@users.noreply.github.com> Date: Thu, 26 Feb 2026 08:18:45 +0900 Subject: [PATCH] fix: rename character_tags to img_character_tags to fix unboundlocalerror --- finetune/tag_images_by_wd14_tagger.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/finetune/tag_images_by_wd14_tagger.py b/finetune/tag_images_by_wd14_tagger.py index b4019819..7840280c 100644 --- a/finetune/tag_images_by_wd14_tagger.py +++ b/finetune/tag_images_by_wd14_tagger.py @@ -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: