diff --git a/invokeai/app/invocations/generate.py b/invokeai/app/invocations/generate.py index 88a76e930c..c9d82ae8de 100644 --- a/invokeai/app/invocations/generate.py +++ b/invokeai/app/invocations/generate.py @@ -181,6 +181,8 @@ class InpaintInvocation(BaseInvocation): device = context.services.model_manager.mgr.cache.execution_device dtype = context.services.model_manager.mgr.cache.precision + vae.to(dtype=unet.dtype) + pipeline = StableDiffusionGeneratorPipeline( vae=vae, text_encoder=None,