From 157c25f123bfa24aa5d72d700016de5642fcb715 Mon Sep 17 00:00:00 2001
From: Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com>
Date: Sun, 2 Apr 2023 17:41:30 +0800
Subject: [PATCH] Restore type

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

diff --git a/modules/sd_vae.py b/modules/sd_vae.py
index 9b00f76e9..707d1fb2c 100644
--- a/modules/sd_vae.py
+++ b/modules/sd_vae.py
@@ -183,6 +183,8 @@ unspecified = object()
 def reload_vae_weights(sd_model=None, vae_file=unspecified):
     from modules import lowvram, devices, sd_hijack
 
+    if devices.dtype_vae == torch.bfloat16:
+        devices.dtype_vae = torch.float16
     if not sd_model:
         sd_model = shared.sd_model