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

Don't crash CLI on exceptions (#2066)

Browse Source
This commit is contained in:
Damian Stewart 2022-12-18 16:28:47 +01:00 committed by GitHub
parent f41da11d66
commit 147834e99c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/generate.py
View File

@ -545,7 +545,7 @@ class Generate:
print('**Interrupted** Partial results will be returned.')
else:
raise KeyboardInterrupt
except RuntimeError as e:
except (RuntimeError, Exception) as e:
print(traceback.format_exc(), file=sys.stderr)
print('>> Could not generate image.')