From 7c6b237fe89d074510da0cd4f382a07829649547 Mon Sep 17 00:00:00 2001 From: John Pollock Date: Sun, 19 Apr 2026 21:05:07 -0500 Subject: [PATCH] Match Copilot image_size restore pattern --- 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 2b712d113..61d3532a1 100644 --- a/comfy_extras/nodes_trellis2.py +++ b/comfy_extras/nodes_trellis2.py @@ -256,8 +256,8 @@ def run_conditioning(model, cropped_img_tensor, include_1024=True): model_internal = model.model device = comfy.model_management.intermediate_device() torch_device = comfy.model_management.get_torch_device() - image_size_missing = object() - original_image_size = getattr(model_internal, "image_size", image_size_missing) + had_image_size = hasattr(model_internal, "image_size") + original_image_size = getattr(model_internal, "image_size", None) def prepare_tensor(pil_img, size): resized_pil = pil_img.resize((size, size), Image.Resampling.LANCZOS) @@ -276,7 +276,7 @@ def run_conditioning(model, cropped_img_tensor, include_1024=True): input_1024 = prepare_tensor(cropped_img_tensor, 1024) cond_1024 = model_internal(input_1024, skip_norm_elementwise=True)[0] finally: - if original_image_size is image_size_missing: + if not had_image_size: delattr(model_internal, "image_size") else: model_internal.image_size = original_image_size