From 3a83168f8734dbaed468f1b7613cc634f30f6200 Mon Sep 17 00:00:00 2001 From: N00byKing Date: Sun, 7 Sep 2025 19:44:26 +0200 Subject: [PATCH] Fix vae load with --cpu-vae --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index cb6580f73..a2b086f9a 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -913,7 +913,7 @@ def vae_dtype(device=None, allowed_dtypes=[]): return torch.float16 elif args.bf16_vae: return torch.bfloat16 - elif args.fp32_vae: + elif args.fp32_vae or (device and is_device_cpu(device)): return torch.float32 for d in allowed_dtypes: