diff --git a/ldm/generate.py b/ldm/generate.py
index db36717135..1b1edabe4d 100644
--- a/ldm/generate.py
+++ b/ldm/generate.py
@@ -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.')