Merge pull request #14775 from AUTOMATIC1111/fix-CLIP-Interrogator-topN-regex

fix CLIP Interrogator topN regex

Merge pull request #14775 from AUTOMATIC1111/fix-CLIP-Interrogator-topN-regex
fix CLIP Interrogator topN regex
0a3a8338 · AUTOMATIC1111 · GitHub · 7ea08593 · 486aeda3 · 0a3a8338
Commit 0a3a8338 authored Jan 27, 2024 by AUTOMATIC1111 Committed by GitHub Jan 27, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/interrogate.py modules/interrogate.py +1 -1

No files found.
--- 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')]