From a9518f0293ce81465a2dfb6e345556271f2a972d Mon Sep 17 00:00:00 2001
From: "Yousef R. Gamaleldin" <81116377+yousef-rafat@users.noreply.github.com>
Date: Thu, 14 May 2026 14:54:07 +0300
Subject: [PATCH] Update comfy_extras/nodes_trellis2.py

Co-authored-by: Alexis Rolland <alexis@comfy.org>
---
 comfy_extras/nodes_trellis2.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/comfy_extras/nodes_trellis2.py b/comfy_extras/nodes_trellis2.py
index 407ac105b..24ac373e2 100644
--- a/comfy_extras/nodes_trellis2.py
+++ b/comfy_extras/nodes_trellis2.py
@@ -1844,9 +1844,9 @@ class Trellis2Extension(ComfyExtension):
             EmptyShapeLatentTrellis2,
             EmptyStructureLatentTrellis2,
             EmptyTextureLatentTrellis2,
-            VaeDecodeTextureTrellis,
-            VaeDecodeShapeTrellis,
-            VaeDecodeStructureTrellis2,
+            VaeDecodeTrellis2Texture,
+            VaeDecodeTrellis2Shape,
+            VaeDecodeTrellis2Structure,
             Trellis2UpsampleCascade,
             PostProcessMesh
         ]