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

fixes for B007

Browse Source
This commit is contained in:
AUTOMATIC
2023-05-10 11:37:18 +03:00
parent 550256db1c
commit a5121e7a06
28 changed files with 57 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/interrogate.py
View File

@@ -207,8 +207,8 @@ class InterrogateModels:
image_features /= image_features.norm(dim=-1, keepdim=True)
for name, topn, items in self.categories():
matches = self.rank(image_features, items, top_count=topn)
for cat in self.categories():
matches = self.rank(image_features, cat.items, top_count=cat.topn)
for match, score in matches:
if shared.opts.interrogate_return_ranks:
res += f", ({match}:{score/100:.3f})"