Skip to content

Commit 070344c

Browse files
committed
Don't crash CLI on exceptions
1 parent b035ac8 commit 070344c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ldm/generate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -545,7 +545,7 @@ def process_image(image,seed):
545545
print('**Interrupted** Partial results will be returned.')
546546
else:
547547
raise KeyboardInterrupt
548-
except RuntimeError as e:
548+
except (RuntimeError, Exception) as e:
549549
print(traceback.format_exc(), file=sys.stderr)
550550
print('>> Could not generate image.')
551551

0 commit comments

Comments
 (0)