diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py index 3d3dfde3b8..9698687157 100644 --- a/ldm/simplet2i.py +++ b/ldm/simplet2i.py @@ -280,7 +280,7 @@ class T2I: tic = time.time() if torch.cuda.is_available(): - torch.cuda.torch.cuda.reset_peak_memory_stats() + torch.cuda.reset_peak_memory_stats() results = list() try: