From d43333ff7193279acee5d7284b40902af351c9d7 Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Wed, 30 Aug 2023 21:13:24 +0300
Subject: [PATCH] fix an issue where VAE would remain in fp16 after an
 auto-switch to fp32

---
 modules/sd_samplers_common.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py
index 6c935a38f..58efcad23 100644
--- a/modules/sd_samplers_common.py
+++ b/modules/sd_samplers_common.py
@@ -95,6 +95,8 @@ def images_tensor_to_samples(image, approximation=None, model=None):
     else:
         if model is None:
             model = shared.sd_model
+        model.first_stage_model.to(devices.dtype_vae)
+
         image = image.to(shared.device, dtype=devices.dtype_vae)
         image = image * 2 - 1
         if len(image) > 1: