diff --git a/invokeai/app/invocations/generate.py b/invokeai/app/invocations/generate.py index 88a76e930cb..c9d82ae8de3 100644 --- a/invokeai/app/invocations/generate.py +++ b/invokeai/app/invocations/generate.py @@ -181,6 +181,8 @@ def _lora_loader(): 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,