298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-15 14:09:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update comfy_extras/nodes_mask.py

Browse Source 
Co-authored-by: missionfloyd <missionfloyd@users.noreply.github.com>
This commit is contained in:
comfyanonymous 2023-04-14 00:12:15 -04:00 committed by GitHub
parent 022a9f271b
commit 35a2c790b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy_extras/nodes_mask.py
View File

@ -9,8 +9,8 @@ class LatentCompositeMasked:
"required": { "required": {
"destination": ("LATENT",), "destination": ("LATENT",),
"source": ("LATENT",), "source": ("LATENT",),
"x": ("INT", {"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 8}), "x": ("INT", {"default": 0, "min": -MAX_RESOLUTION, "max": MAX_RESOLUTION, "step": 8}),
"y": ("INT", {"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 8}), "y": ("INT", {"default": 0, "min": -MAX_RESOLUTION, "max": MAX_RESOLUTION, "step": 8}),
}, },
"optional": { "optional": {
"mask": ("MASK",), "mask": ("MASK",),
@ -26,6 +26,9 @@ class LatentCompositeMasked:
destination = destination["samples"].clone() destination = destination["samples"].clone()
source = source["samples"] source = source["samples"]
x = max(-source.shape[3] * 8, min(x, destination.shape[3] * 8))
y = max(-source.shape[2] * 8, min(y, destination.shape[2] * 8))
left, top = (x // 8, y // 8) left, top = (x // 8, y // 8)
right, bottom = (left + source.shape[3], top + source.shape[2],) right, bottom = (left + source.shape[3], top + source.shape[2],)
@ -40,7 +43,7 @@ class LatentCompositeMasked:
# calculate the bounds of the source that will be overlapping the destination # calculate the bounds of the source that will be overlapping the destination
# this prevents the source trying to overwrite latent pixels that are out of bounds # this prevents the source trying to overwrite latent pixels that are out of bounds
# of the destination # of the destination
visible_width, visible_height = (destination.shape[3] - left, destination.shape[2] - top,) visible_width, visible_height = (destination.shape[3] - left + min(0, x), destination.shape[2] - top + min(0, y),)
mask = mask[:, :, :visible_height, :visible_width] mask = mask[:, :, :visible_height, :visible_width]
inverse_mask = torch.ones_like(mask) - mask inverse_mask = torch.ones_like(mask) - mask