hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 05:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source 
fix CLIP Interrogator topN regex
This commit is contained in:
AUTOMATIC1111
2024-01-27 14:36:27 +03:00
committed by GitHub
parent 7ea0859362 486aeda3a7
commit 0a3a83382f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/interrogate.py
View File

@@ -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')]