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

fix run_conditioning

Browse Source
This commit is contained in:
Yousef Rafat 2026-02-18 21:52:40 +02:00
parent c14317d6e0
commit ff04ef5558
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy_extras/nodes_trellis2.py
View File

@ -71,7 +71,14 @@ def smart_crop_square(image, mask, margin_ratio=0.1, bg_color=(128, 128, 128)):
dst_x2 = dst_x1 + (src_x2 - src_x1)
dst_y2 = dst_y1 + (src_y2 - src_y1)
canvas[:, dst_y1:dst_y2, dst_x1:dst_x2] = image[:, src_y1:src_y2, src_x1:src_x2]
img_crop = image[:, src_y1:src_y2, src_x1:src_x2]
mask_crop = mask[0, src_y1:src_y2, src_x1:src_x2]
bg_val = torch.tensor(bg_color, device=image.device, dtype=image.dtype).view(-1, 1, 1) / 255.0
masked_crop = img_crop * mask_crop + bg_val * (1.0 - mask_crop)
canvas[:, dst_y1:dst_y2, dst_x1:dst_x2] = masked_crop
return canvas