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.
This commit is contained in:
Lincoln Stein 2022-09-01 10:23:45 -04:00 committed by GitHub
parent 2cac4697aa
commit 2455bb38a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -280,7 +280,7 @@ class T2I:
tic = time.time() tic = time.time()
if torch.cuda.is_available(): if torch.cuda.is_available():
torch.cuda.torch.cuda.reset_peak_memory_stats() torch.cuda.reset_peak_memory_stats()
results = list() results = list()
try: try: