diff options
| author | Greendayle <Greendayle> | 2022-10-05 19:15:08 +0000 |
|---|---|---|
| committer | Greendayle <Greendayle> | 2022-10-05 19:15:08 +0000 |
| commit | 1506fab29ad54beb9f52236912abc432209c8089 (patch) | |
| tree | 0f8d0c66955345c0a63823ef0fc91ae5be83aa70 /modules/deepbooru.py | |
| parent | 59a2b9e5afc27d2fda72069ca0635070535d18fe (diff) | |
| download | stable-diffusion-webui-gfx803-1506fab29ad54beb9f52236912abc432209c8089.tar.gz stable-diffusion-webui-gfx803-1506fab29ad54beb9f52236912abc432209c8089.tar.bz2 stable-diffusion-webui-gfx803-1506fab29ad54beb9f52236912abc432209c8089.zip | |
removing problematic tag
Diffstat (limited to 'modules/deepbooru.py')
| -rw-r--r-- | modules/deepbooru.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/modules/deepbooru.py b/modules/deepbooru.py index 958b1c3d..841cb9c5 100644 --- a/modules/deepbooru.py +++ b/modules/deepbooru.py @@ -38,13 +38,12 @@ def _load_tf_and_return_tags(pil_image, threshold): for i, tag in enumerate(tags): result_dict[tag] = y[i] - - - result_tags_out = [] result_tags_print = [] for tag in tags: if result_dict[tag] >= threshold: + if tag.startswith("rating:"): + continue result_tags_out.append(tag) result_tags_print.append(f'{result_dict[tag]} {tag}') |