Restore type

2026-09-19 09:14:35 +02:00 · 2023-04-02 17:41:30 +08:00
parent 4aa115d5c1
commit 157c25f123
1 changed files with 2 additions and 0 deletions
@@ -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