Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor CUDA cache clearing to add statistical reporting. (#2553)

Browse Source
This commit is contained in:
Jonathan
2023-02-06 11:53:30 -06:00
committed by GitHub
parent 8ab66a211c
commit 28b40bebbe
3 changed files with 36 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ldm/invoke/generator/txt2img2img.py
View File

@ -66,8 +66,9 @@ class Txt2Img2Img(Generator):
)
# Free up memory from the last generation.
if self.model.device.type == 'cuda':
torch.cuda.empty_cache()
clear_cuda_cache = kwargs['clear_cuda_cache'] or None
if clear_cuda_cache is not None:
clear_cuda_cache()
second_pass_noise = self.get_noise_like(resized_latents)