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

prevent crashes on malformed models

Browse Source
This commit is contained in:
Lincoln Stein
2023-07-01 14:32:58 -04:00
parent 96212bb35f
commit f1928d2588
4 changed files with 619 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/model_management/model_manager.py
View File

@ -52,7 +52,7 @@ A typical example is:
sd1_5 = mgr.get_model('stable-diffusion-v1-5',
model_type=ModelType.Main,
base_model=BaseModelType.StableDiffusion1,
submodel_type=SubModelType.Unet)
submodel_type=SubModelType.UNet)
with sd1_5 as unet:
run_some_inference(unet)
@ -688,6 +688,7 @@ class ModelManager(object):
model_class = MODEL_CLASSES[cur_base_model][cur_model_type]
if model_class.save_to_config:
model_config.error = ModelError.NotFound
self.models.pop(model_key, None)
else:
self.models.pop(model_key, None)
else: