wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-04 05:17:50 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Better VAE error

Browse Source
This commit is contained in:
kijai 2026-06-01 14:45:57 +03:00
parent 362d3faa95
commit ee9a1ffe19
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/sd.py
View File

@ -901,9 +901,10 @@ class VAE:
self.latent_dim = 1
self.working_dtypes = [torch.float16, torch.bfloat16, torch.float32]
# The generic VAE.encode/decode path isn't used: VAEDecodeTripoSplat calls the gaussian
# decoder directly (Gaussian output, not pixels) and reserves VRAM itself from num_gaussians.
self.memory_used_encode = lambda shape, dtype: 0
self.memory_used_decode = lambda shape, dtype: 0
# decoder directly (structured GaussianSplat objects, not a tensor and reserves VRAM itself from num_gaussians.
def _no_generic_io(*args, **kwargs):
raise RuntimeError("TripoSplat gaussian decoder: use the 'TripoSplat Decode' (VAEDecodeTripoSplat)")
self.memory_used_encode = self.memory_used_decode = _no_generic_io
else:
logging.warning("WARNING: No VAE weights detected, VAE not initalized.")
self.first_stage_model = None