closes #273, crash on M1 machines

lstein · lstein · commit dc30adfbb435 · 2022-09-01T01:09:56.000-04:00
diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py
@@ -279,7 +279,7 @@ def process_image(image,seed):
             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
@@ -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:

Original file line number	Diff line number	Diff line change
`@@ -88,7 +88,7 @@ def main():`
`88`	`88`	`tic = time.time()`
`89`	`89`	`t2i.load_model()`
`90`	`90`	`print(`
`91`		`- f'model loaded in', '%4.2fs' % (time.time() - tic)`
	`91`	`+ f'>> model loaded in', '%4.2fs' % (time.time() - tic)`
`92`	`92`	`)`
`93`	`93`
`94`	`94`	`if not infile:`