diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-12-24 09:40:32 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-12-24 13:22:47 +0000 |
| commit | 03d7b394539558f6f560155d87a4fc66eb675e30 (patch) | |
| tree | 26d27dec0b8f4b4b0273f0a173372d71c518e2e6 | |
| parent | ccaacb1891f018d380880ea4deb4ab79ad33d0d9 (diff) | |
| download | stable-diffusion-webui-gfx803-03d7b394539558f6f560155d87a4fc66eb675e30.tar.gz stable-diffusion-webui-gfx803-03d7b394539558f6f560155d87a4fc66eb675e30.tar.bz2 stable-diffusion-webui-gfx803-03d7b394539558f6f560155d87a4fc66eb675e30.zip | |
added an option to filter out deepbooru tags
| -rw-r--r-- | modules/deepbooru.py | 4 | ||||
| -rw-r--r-- | modules/shared.py | 1 |
2 files changed, 4 insertions, 1 deletions
diff --git a/modules/deepbooru.py b/modules/deepbooru.py index dfc83357..122fce7f 100644 --- 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: diff --git a/modules/shared.py b/modules/shared.py index 8ea3b441..a75de535 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -386,6 +386,7 @@ options_templates.update(options_section(('interrogate', "Interrogate Options"), "deepbooru_sort_alpha": OptionInfo(True, "Interrogate: deepbooru sort alphabetically"),
"deepbooru_use_spaces": OptionInfo(False, "use spaces for tags in deepbooru"),
"deepbooru_escape": OptionInfo(True, "escape (\\) brackets in deepbooru (so they are used as literal brackets and not for emphasis)"),
+ "deepbooru_filter_tags": OptionInfo("", "filter out those tags from deepbooru output (separated by comma)"),
}))
options_templates.update(options_section(('ui', "User interface"), {
|