diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py
index 230aa74c28..9251248994 100644
--- a/ldm/simplet2i.py
+++ b/ldm/simplet2i.py
@@ -279,7 +279,7 @@ class T2I:
             self._set_sampler()
 
         tic = time.time()
-        torch.cuda.torch.cuda.reset_peak_memory_stats()
+        torch.cuda.reset_peak_memory_stats() if self.device == 'cuda' else None
         results = list()
 
         try:
diff --git a/scripts/dream.py b/scripts/dream.py
index 2911e8847a..85b2ed5211 100755
--- a/scripts/dream.py
+++ b/scripts/dream.py
@@ -88,7 +88,7 @@ def main():
     tic = time.time()
     t2i.load_model()
     print(
-        f'model loaded in', '%4.2fs' % (time.time() - tic)
+        f'>> model loaded in', '%4.2fs' % (time.time() - tic)
     )
 
     if not infile: