diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2024-01-27 10:34:07 +0000 |
|---|---|---|
| committer | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2024-01-27 10:35:07 +0000 |
| commit | 486aeda3a75ed20e75d85d6ae6845d66d1cbe1de (patch) | |
| tree | ffa671eac1c680331891d34eb70b9149b5a04937 /modules/interrogate.py | |
| parent | 4d5db58a3e82b441fa0ca15df0370d49175e2aea (diff) | |
| download | stable-diffusion-webui-gfx803-486aeda3a75ed20e75d85d6ae6845d66d1cbe1de.tar.gz stable-diffusion-webui-gfx803-486aeda3a75ed20e75d85d6ae6845d66d1cbe1de.tar.bz2 stable-diffusion-webui-gfx803-486aeda3a75ed20e75d85d6ae6845d66d1cbe1de.zip | |
fix CLIP Interrogator topN regex
Co-Authored-By: Martin Rizzo <60830236+martin-rizzo@users.noreply.github.com>
Diffstat (limited to 'modules/interrogate.py')
| -rw-r--r-- | modules/interrogate.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/interrogate.py b/modules/interrogate.py index 35a627ca..c93e7aa8 100644 --- a/modules/interrogate.py +++ b/modules/interrogate.py @@ -17,7 +17,7 @@ clip_model_name = 'ViT-L/14' Category = namedtuple("Category", ["name", "topn", "items"])
-re_topn = re.compile(r"\.top(\d+)\.")
+re_topn = re.compile(r"\.top(\d+)$")
def category_types():
return [f.stem for f in Path(shared.interrogator.content_dir).glob('*.txt')]
|