From ee9a1ffe199ed10e2dd5ba82171ed1d909bdcf6a Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 1 Jun 2026 14:45:57 +0300
Subject: [PATCH] Better VAE error

---
 comfy/sd.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/comfy/sd.py b/comfy/sd.py
index a46c81293..9a2d31930 100644
--- a/comfy/sd.py
+++ b/comfy/sd.py
@@ -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