From aab6369ffe27552916062b8ce3ca2e591a890d45 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sun, 10 Dec 2023 12:24:50 -0500 Subject: [PATCH] Update invokeai/backend/model_manager/search.py Co-authored-by: Ryan Dick --- invokeai/backend/model_manager/search.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/invokeai/backend/model_manager/search.py b/invokeai/backend/model_manager/search.py index 45019bb103..4cc3caebe4 100644 --- a/invokeai/backend/model_manager/search.py +++ b/invokeai/backend/model_manager/search.py @@ -128,11 +128,7 @@ class ModelSearch(ModelSearchBase): def model_found(self, model: Path) -> None: self.stats.models_found += 1 - if not self.on_model_found: - self.stats.models_filtered += 1 - self.models_found.add(model) - return - if self.on_model_found(model): + if not self.on_model_found or self.on_model_found(model): self.stats.models_filtered += 1 self.models_found.add(model)