diff --git a/ldm/generate.py b/ldm/generate.py index 1b1edabe4d..db36717135 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, Exception) as e: + except RuntimeError as e: print(traceback.format_exc(), file=sys.stderr) print('>> Could not generate image.')