xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-15 14:30:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix vae load with --cpu-vae

Browse Source
This commit is contained in:
N00byKing 2025-09-07 19:44:26 +02:00
parent bcbd7884e3
commit 3a83168f87
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -913,7 +913,7 @@ def vae_dtype(device=None, allowed_dtypes=[]):
return torch.float16 return torch.float16
elif args.bf16_vae: elif args.bf16_vae:
return torch.bfloat16 return torch.bfloat16
elif args.fp32_vae: elif args.fp32_vae or (device and is_device_cpu(device)):
return torch.float32 return torch.float32
for d in allowed_dtypes: for d in allowed_dtypes: