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

stability improvements

Browse Source 
- provide full traceback when a model fails to load
- fix VAE record for VoxelArt; otherwise load fails
This commit is contained in:
Lincoln Stein
2023-01-23 21:40:27 -05:00
parent d7554b01fd
commit bc2194228e
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
configs/INITIAL_MODELS.yaml
View File

@ -93,6 +93,7 @@ voxel_art-1.0:
format: ckpt format: ckpt
vae: vae:
repo_id: stabilityai/sd-vae-ft-mse repo_id: stabilityai/sd-vae-ft-mse
file: vae-ft-mse-840000-ema-pruned.ckpt
recommended: False recommended: False
width: 512 width: 512
height: 512 height: 512
@ -102,7 +103,7 @@ ft-mse-improved-autoencoder-840000:
format: ckpt format: ckpt
config: VAE/default config: VAE/default
file: vae-ft-mse-840000-ema-pruned.ckpt file: vae-ft-mse-840000-ema-pruned.ckpt
recommended: False recommended: True
width: 512 width: 512
height: 512 height: 512
trinart_vae: trinart_vae:

1
ldm/generate.py
View File

@ -844,6 +844,7 @@ class Generate:
model_data = cache.get_model(model_name) model_data = cache.get_model(model_name)
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)}')
print(traceback.format_exc(), file=sys.stderr)
if previous_model_name is None: if previous_model_name is None:
raise e raise e
print(f'** trying to reload previous model') print(f'** trying to reload previous model')