Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update invokeai/backend/model_manager/search.py

Browse Source 
Co-authored-by: Ryan Dick <ryanjdick3@gmail.com>
This commit is contained in:
Lincoln Stein 2023-12-10 12:24:50 -05:00 committed by GitHub
parent 913c68982a
commit aab6369ffe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/model_manager/search.py
View File

@ -128,11 +128,7 @@ class ModelSearch(ModelSearchBase):
def model_found(self, model: Path) -> None: def model_found(self, model: Path) -> None:
self.stats.models_found += 1 self.stats.models_found += 1
if not self.on_model_found: if not self.on_model_found or self.on_model_found(model):
self.stats.models_filtered += 1
self.models_found.add(model)
return
if self.on_model_found(model):
self.stats.models_filtered += 1 self.stats.models_filtered += 1
self.models_found.add(model) self.models_found.add(model)