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