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

Just comment out cache emptying

Browse Source
This commit is contained in:
Brandon Rising 2024-02-07 14:47:31 -05:00
parent db5f1c8623
commit 52568c7329
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/app/invocations/latent.py
View File

@ -860,9 +860,9 @@ class LatentsToImageInvocation(BaseInvocation, WithMetadata):
vae.disable_tiling() vae.disable_tiling()
# clear memory as vae decode can request a lot # clear memory as vae decode can request a lot
torch.cuda.empty_cache() # torch.cuda.empty_cache()
if choose_torch_device() == torch.device("mps"): # if choose_torch_device() == torch.device("mps"):
mps.empty_cache() # mps.empty_cache()
with torch.inference_mode(): with torch.inference_mode():
# copied from diffusers pipeline # copied from diffusers pipeline
@ -874,9 +874,9 @@ class LatentsToImageInvocation(BaseInvocation, WithMetadata):
image = VaeImageProcessor.numpy_to_pil(np_image)[0] image = VaeImageProcessor.numpy_to_pil(np_image)[0]
torch.cuda.empty_cache() # torch.cuda.empty_cache()
if choose_torch_device() == torch.device("mps"): # if choose_torch_device() == torch.device("mps"):
mps.empty_cache() # mps.empty_cache()
image_dto = context.services.images.create( image_dto = context.services.images.create(
image=image, image=image,