added an option to filter out deepbooru tags

2026-04-27 01:38:53 +00:00 · 2022-12-24 12:40:32 +03:00
parent ccaacb1891
commit 03d7b39453
2 changed files with 4 additions and 1 deletions
--- a/modules/deepbooru.py
+++ b/modules/deepbooru.py
@@ -79,7 +79,9 @@ class DeepDanbooru:

        res = []

-        for tag in tags:
+        filtertags = set([x.strip().replace(' ', '_') for x in shared.opts.deepbooru_filter_tags.split(",")])
+
+        for tag in [x for x in tags if x not in filtertags]:
            probability = probability_dict[tag]
            tag_outformat = tag
            if use_spaces: