xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-14 07:14:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update image_nodes.py

Browse Source
This commit is contained in:
kijai 2025-04-02 02:47:03 +03:00
parent 59bd92ff49
commit 18a5565b5c
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nodes/image_nodes.py
View File

@ -1868,10 +1868,11 @@ Inserts empty frames between the images in a batch.
"required": { "required": {
"images": ("IMAGE",), "images": ("IMAGE",),
"empty_frames_per_image": ("INT", {"default": 1,"min": 0, "max": 4096, "step": 1}), "empty_frames_per_image": ("INT", {"default": 1,"min": 0, "max": 4096, "step": 1}),
"pad_frame_value": ("FLOAT", {"default": 0.0,"min": 0.0, "max": 1.0, "step": 0.01}),
}, },
} }
def pad(self, images, empty_frames_per_image): def pad(self, images, empty_frames_per_image, pad_frame_value):
B, H, W, C = images.shape B, H, W, C = images.shape
if B == 1 or empty_frames_per_image == 0: if B == 1 or empty_frames_per_image == 0:
@ -1881,9 +1882,9 @@ Inserts empty frames between the images in a batch.
total_frames = B + (B-1) * empty_frames_per_image total_frames = B + (B-1) * empty_frames_per_image
# Create new tensor with zeros (empty frames) # Create new tensor with zeros (empty frames)
padded_batch = torch.zeros((total_frames, H, W, C), padded_batch = torch.ones((total_frames, H, W, C),
dtype=images.dtype, dtype=images.dtype,
device=images.device) device=images.device) * pad_frame_value
# Create mask tensor (1 for original frames, 0 for empty frames) # Create mask tensor (1 for original frames, 0 for empty frames)
mask = torch.zeros((total_frames, H, W), mask = torch.zeros((total_frames, H, W),
dtype=images.dtype, dtype=images.dtype,