OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-03-03 04:24:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix CLIP Interrogator topN regex

Browse Source 
Co-Authored-By: Martin Rizzo <60830236+martin-rizzo@users.noreply.github.com>
This commit is contained in:
w-e-w 2024-01-27 19:34:07 +09:00
parent 4d5db58a3e
commit 486aeda3a7
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')]