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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2023-12-24 18:59:43 +02:00
parent a8c1a707a9
commit 8dbfddc5ec
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
nodes.py
View File

@ -2891,6 +2891,26 @@ class StableZero123_BatchSchedule:
latent = torch.zeros([batch_size, 4, height // 8, width // 8])
return (final_positive, final_negative, {"samples": latent})
class ImageBatchRepeatEveryNth:
RETURN_TYPES = ("IMAGE",)
FUNCTION = "repeat"
CATEGORY = "KJNodes"
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"images": ("IMAGE",),
"repeats": ("INT", {"default": 1, "min": 1, "max": 4096}),
},
}
def repeat(self, images, repeats):
repeated_images = torch.repeat_interleave(images, repeats=repeats, dim=0)
return (repeated_images, )
NODE_CLASS_MAPPINGS = {
"INTConstant": INTConstant,
@ -2945,6 +2965,7 @@ NODE_CLASS_MAPPINGS = {
"SoundReactive": SoundReactive,
"GenerateNoise": GenerateNoise,
"StableZero123_BatchSchedule": StableZero123_BatchSchedule,
"ImageBatchRepeatEveryNth": ImageBatchRepeatEveryNth
}
NODE_DISPLAY_NAME_MAPPINGS = {
"INTConstant": "INT Constant",
@ -2998,4 +3019,5 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"SoundReactive": "SoundReactive",
"GenerateNoise": "GenerateNoise",
"StableZero123_BatchSchedule": "StableZero123_BatchSchedule",
"ImageBatchRepeatEveryNth": "ImageBatchRepeatEveryNth"
}