stability improvements

- 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

View File

@ -844,6 +844,7 @@ class Generate:
model_data = cache.get_model(model_name)
except Exception as 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:
raise e
print(f'** trying to reload previous model')