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,