diff --git a/ldm/generate.py b/ldm/generate.py
index 05e2c6a440..8f67403633 100644
--- a/ldm/generate.py
+++ b/ldm/generate.py
@@ -357,12 +357,14 @@ class Generate:
         print(
             f'>>   {len(results)} image(s) generated in', '%4.2fs' % (toc - tic)
         )
-        print(
-            f'>>   Max VRAM used for this generation:',
-            '%4.2fG' % (torch.cuda.max_memory_allocated() / 1e9),
-        )
+        if torch.cuda.is_available() and self.device.type == 'cuda':
+            print(
+                f'>>   Max VRAM used for this generation:',
+                '%4.2fG.' % (torch.cuda.max_memory_allocated() / 1e9),
+                'Current VRAM utilization:'
+                '%4.2fG' % (torch.cuda.memory_allocated() / 1e9),
+            )
 
-        if self.session_peakmem:
             self.session_peakmem = max(
                 self.session_peakmem, torch.cuda.max_memory_allocated()
             )
diff --git a/scripts/dream.py b/scripts/dream.py
index 0b1e80f66d..ece17d122b 100755
--- a/scripts/dream.py
+++ b/scripts/dream.py
@@ -130,8 +130,11 @@ def main_loop(t2i, outdir, prompt_as_dir, parser, infile):
             command = get_next_command(infile)
         except EOFError:
             done = True
-            break
-
+            continue
+        except KeyboardInterrupt:
+            done = True
+            continue
+        
         # skip empty lines
         if not command.strip():
             continue