Compare commits

...

3 Commits

Author SHA1 Message Date
52568c7329 Just comment out cache emptying 2024-02-07 14:47:31 -05:00
db5f1c8623 Revert "L2I Performance updates"
This reverts commit 8f0352f3ad.
2024-02-07 14:46:53 -05:00
8f0352f3ad L2I Performance updates 2024-02-07 13:01:41 -05:00

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,