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

gracefully recover from failed model load

Browse Source
This commit is contained in:
Lincoln Stein 2022-10-13 12:27:04 -04:00
parent 1c102c71fc
commit 916f5bfbb2
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ldm/invoke/model_cache.py
View File

@ -71,8 +71,9 @@ class ModelCache(object):
self.models[model_name]['height'] = height self.models[model_name]['height'] = height
self.models[model_name]['hash'] = hash self.models[model_name]['hash'] = hash
except Exception as e: except Exception as e:
print(f'** model {model_name} could not be loaded: {str(e)}') print(f'** model {model_name} could not be loaded: {str(e)}')
return {} print(f'** restoring {self.current_model}')
return self.get_model(self.current_model)
self.current_model = model_name self.current_model = model_name
self._push_newest_model(model_name) self._push_newest_model(model_name)