diff --git a/comfy_extras/nodes_trellis2.py b/comfy_extras/nodes_trellis2.py
index b1ad5d1e1..c8ac9bc33 100644
--- a/comfy_extras/nodes_trellis2.py
+++ b/comfy_extras/nodes_trellis2.py
@@ -264,12 +264,12 @@ def run_conditioning(model, cropped_img_tensor, include_1024=True):
         img_t = torch.from_numpy(img_np).permute(2, 0, 1).unsqueeze(0).to(torch_device)
         return (img_t - dino_mean.to(torch_device)) / dino_std.to(torch_device)
 
-    model_internal.image_size = 512
-    input_512 = prepare_tensor(cropped_img_tensor, 512)
-    cond_512 = model_internal(input_512, skip_norm_elementwise=True)[0]
-
     cond_1024 = None
     try:
+        model_internal.image_size = 512
+        input_512 = prepare_tensor(cropped_img_tensor, 512)
+        cond_512 = model_internal(input_512, skip_norm_elementwise=True)[0]
+
         if include_1024:
             model_internal.image_size = 1024
             input_1024 = prepare_tensor(cropped_img_tensor, 1024)