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

do not ignore keyboard interrupt while scanning models

Browse Source
This commit is contained in:
Lincoln Stein
2023-09-30 14:21:39 -04:00
parent c025c9c4ed
commit 230ee18536
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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))