Enable tiling on vae when decoding with low_vram flag on krea2

extensions_built_in/diffusion_models/krea2/krea2.py

@@ -467,7 +467,16 @@ class Krea2Model(BaseModel):
         )
         latents = latents * latents_std + latents_mean
 
+        # Full-resolution decode spikes VRAM; tile it when low on VRAM (decode
+        # only -- encode stays untiled).
+        tiled = self.model_config.low_vram
+        if tiled:
+            self.vae.enable_tiling()
+        try:
             images = self.vae.decode(latents).sample
+        finally:
+            if tiled:
+                self.vae.disable_tiling()
         images = images.squeeze(2)  # drop frame dim
         return images.to(device, dtype=dtype)