diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 08:37:18 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 08:37:18 +0000 |
| commit | a5121e7a0623db328a9462d340d389ed6737374a (patch) | |
| tree | 2fa6051d457555ef2d61793af2756a44c0ea221c /modules/interrogate.py | |
| parent | 550256db1ce18778a9d56ff343d844c61b9f9b83 (diff) | |
| download | stable-diffusion-webui-gfx803-a5121e7a0623db328a9462d340d389ed6737374a.tar.gz stable-diffusion-webui-gfx803-a5121e7a0623db328a9462d340d389ed6737374a.tar.bz2 stable-diffusion-webui-gfx803-a5121e7a0623db328a9462d340d389ed6737374a.zip | |
fixes for B007
Diffstat (limited to 'modules/interrogate.py')
| -rw-r--r-- | modules/interrogate.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/interrogate.py b/modules/interrogate.py index a1c8e537..111b1322 100644 --- a/modules/interrogate.py +++ b/modules/interrogate.py @@ -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})"
|