From 2455bb38a479c6dacdf2f459fcb673a951197648 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 1 Sep 2022 10:23:45 -0400 Subject: [PATCH] Remove redundant chain of types torch->cuda and cuda->torch, so torch.cuda.torch.cuda actually works. However it looks like (and probably is) a typo. --- ldm/simplet2i.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: