wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-27 19:02:31 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Match Copilot image_size restore pattern

Browse Source
This commit is contained in:
John Pollock 2026-04-19 21:05:07 -05:00
parent 2ad1ca5531
commit 7c6b237fe8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes_trellis2.py
View File

@ -256,8 +256,8 @@ def run_conditioning(model, cropped_img_tensor, include_1024=True):
model_internal = model.model model_internal = model.model
device = comfy.model_management.intermediate_device() device = comfy.model_management.intermediate_device()
torch_device = comfy.model_management.get_torch_device() torch_device = comfy.model_management.get_torch_device()
image_size_missing = object() had_image_size = hasattr(model_internal, "image_size")
original_image_size = getattr(model_internal, "image_size", image_size_missing) original_image_size = getattr(model_internal, "image_size", None)
def prepare_tensor(pil_img, size): def prepare_tensor(pil_img, size):
resized_pil = pil_img.resize((size, size), Image.Resampling.LANCZOS) 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) input_1024 = prepare_tensor(cropped_img_tensor, 1024)
cond_1024 = model_internal(input_1024, skip_norm_elementwise=True)[0] cond_1024 = model_internal(input_1024, skip_norm_elementwise=True)[0]
finally: finally:
if original_image_size is image_size_missing: if not had_image_size:
delattr(model_internal, "image_size") delattr(model_internal, "image_size")
else: else:
model_internal.image_size = original_image_size model_internal.image_size = original_image_size