Merge branch 'master' of github.com:toriato/stable-diffusion-webui-wd…

…14-tagger
toriato · Jan 3, 2023 · 3b8f68a · 3b8f68a
2 parents d5ea1aa + 988aaab
commit 3b8f68a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/install.py b/install.py
diff --git a/tagger/interrogator.py b/tagger/interrogator.py
@@ -70,6 +70,7 @@ def postprocess_tags(
             )
         }
 
+        new_tags = []
         for tag in list(tags):
             new_tag = tag
 
@@ -82,8 +83,8 @@ def postprocess_tags(
             if add_confident_as_weight:
                 new_tag = f'({new_tag}:{tags[tag]})'
 
-            if new_tag != tag:
-                tags[new_tag] = tags.pop(tag)
+            new_tags.append((new_tag, tags[tag]))
+        tags = dict(new_tags)
 
         return tags