wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-02 05:12:36 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Handle missing Trellis2 image_size restore state

Browse Source
This commit is contained in:
John Pollock 2026-04-19 20:51:22 -05:00
parent d7416e5690
commit 2ad1ca5531
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_trellis2.py
View File

@ -256,7 +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()
original_image_size = getattr(model_internal, "image_size", None) image_size_missing = object()
original_image_size = getattr(model_internal, "image_size", image_size_missing)
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)
@ -275,7 +276,10 @@ 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:
model_internal.image_size = original_image_size if original_image_size is image_size_missing:
delattr(model_internal, "image_size")
else:
model_internal.image_size = original_image_size
conditioning = { conditioning = {
'cond_512': cond_512.to(device), 'cond_512': cond_512.to(device),