OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-03-09 15:34:54 +08:00
Code Issues 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
commit 0a3a83382f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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')]