From e5be980ebe5df2b542061e641a622e2bb90690e3 Mon Sep 17 00:00:00 2001
From: Sergey Borisov <stalkek7779@yandex.ru>
Date: Thu, 10 Aug 2023 23:17:09 +0300
Subject: [PATCH] Temporary force set vae to same precision as unet

---
 invokeai/app/invocations/generate.py | 2 ++
 1 file changed, 2 insertions(+)

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,