From a8cc5caf969e30dcf8cd3df6c5218256ecc0e29d Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Fri, 19 Jul 2024 17:07:52 -0400 Subject: [PATCH] Fix function call that we forgot to update in https://github.com/invoke-ai/InvokeAI/pull/6606 --- .../app/invocations/tiled_multi_diffusion_denoise_latents.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invokeai/app/invocations/tiled_multi_diffusion_denoise_latents.py b/invokeai/app/invocations/tiled_multi_diffusion_denoise_latents.py index edb7691e4d..409171794e 100644 --- a/invokeai/app/invocations/tiled_multi_diffusion_denoise_latents.py +++ b/invokeai/app/invocations/tiled_multi_diffusion_denoise_latents.py @@ -222,7 +222,8 @@ class TiledMultiDiffusionDenoiseLatents(BaseInvocation): context=context, positive_conditioning_field=self.positive_conditioning, negative_conditioning_field=self.negative_conditioning, - unet=unet, + device=unet.device, + dtype=unet.dtype, latent_height=latent_tile_height, latent_width=latent_tile_width, cfg_scale=self.cfg_scale,