upstream-archive/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-09-21 05:27:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix ace step VAE decode crashing on non bf16 GPUs. (#16405)

Browse Source
This commit is contained in:
comfyanonymous
2026-09-18 14:35:35 -04:00
committed by GitHub
parent 0d9017220a
commit 3c80da7f87
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/sd.py
+3
View File
@@ -1095,6 +1095,9 @@ class VAE:
self.patcher = mp(self.first_stage_model, load_device=self.device, offload_device=offload_device, fast_disk=fast_disk)
m, u = self.first_stage_model.load_state_dict(sd, strict=False, assign=self.patcher.is_dynamic())
if not self.patcher.is_dynamic():
# Lazy parameters only exist after loading the state dict.
self.first_stage_model.to(self.vae_dtype)
if len(m) > 0:
logging.warning("Missing VAE keys {}".format(m))