From 230ee18536653563a27b8ba44d66d64db43cd4f9 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sat, 30 Sep 2023 14:21:39 -0400 Subject: [PATCH] do not ignore keyboard interrupt while scanning models --- invokeai/backend/model_manager/search.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/invokeai/backend/model_manager/search.py b/invokeai/backend/model_manager/search.py index 0670ab2138..4314249cbb 100644 --- a/invokeai/backend/model_manager/search.py +++ b/invokeai/backend/model_manager/search.py @@ -180,6 +180,8 @@ class ModelSearch(ModelSearchBase): self._scanned_dirs.add(path) try: self.model_found(path) + except KeyboardInterrupt: + raise except Exception as e: self.logger.warning(str(e)) @@ -190,5 +192,7 @@ class ModelSearch(ModelSearchBase): if path.suffix in {".ckpt", ".bin", ".pth", ".safetensors", ".pt"}: try: self.model_found(path) + except KeyboardInterrupt: + raise except Exception as e: self.logger.warning(str(e))