From 614b167994aa847fda2b4a718048a2cdbc84c132 Mon Sep 17 00:00:00 2001
From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com>
Date: Mon, 2 Feb 2026 21:23:19 +0200
Subject: [PATCH] .

---
 comfy_extras/nodes_trellis2.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/comfy_extras/nodes_trellis2.py b/comfy_extras/nodes_trellis2.py
index 304d95493..70bbbb29d 100644
--- a/comfy_extras/nodes_trellis2.py
+++ b/comfy_extras/nodes_trellis2.py
@@ -206,7 +206,8 @@ class EmptyShapeLatentTrellis2(IO.ComfyNode):
                 IO.Latent.Output(),
             ]
         )
-    
+
+    @classmethod
     def execute(cls, structure_output):
         # i will see what i have to do here
         coords = structure_output or structure_output.coords
@@ -227,7 +228,8 @@ class EmptyTextureLatentTrellis2(IO.ComfyNode):
                 IO.Latent.Output(),
             ]
         )
-    
+
+    @classmethod
     def execute(cls, structure_output):
         # TODO
         in_channels = 32
@@ -248,7 +250,7 @@ class EmptyStructureLatentTrellis2(IO.ComfyNode):
                 IO.Latent.Output(),
             ]
         )
-    
+    @classmethod
     def execute(cls, res, batch_size):
         in_channels = 32
         latent = torch.randn(batch_size, in_channels, res, res, res)