Added support for multiple images as RefImages

This commit is contained in:
Sirshendu Ganguly
2026-04-24 19:17:14 +05:30
parent 47b637bd26
commit c30cbd5bb2
7 changed files with 88 additions and 25 deletions
@@ -0,0 +1,46 @@
from typing import List
from .utils import runwareUtils as rwUtils
class RunwareVideoInputsReferenceImagesMultipleImagesConnector:
"""Collect up to 4 images for one referenceImages entry."""
MAX_IMAGES = 4
@classmethod
def INPUT_TYPES(cls):
optionalInputs = {}
for i in range(1, cls.MAX_IMAGES + 1):
optionalInputs[f"Image{i}"] = ("IMAGE", {
"tooltip": f"Optional image {i} for this reference group.",
})
return {
"required": {},
"optional": optionalInputs,
}
DESCRIPTION = (
"Collect up to 4 images for one referenceImages item. "
"Connect this output to Images1/Images2/... on Runware Video Inference Inputs Reference Images."
)
FUNCTION = "createImages"
RETURN_TYPES = ("RUNWAREVIDEOINPUTSREFERENCEMULTIIMAGES",)
RETURN_NAMES = ("Images",)
CATEGORY = "Runware"
def createImages(self, **kwargs) -> tuple[List[str]]:
images: List[str] = []
for i in range(1, self.MAX_IMAGES + 1):
image = kwargs.get(f"Image{i}")
if image is not None:
images.append(rwUtils.convertTensor2IMG(image))
return (images,)
NODE_CLASS_MAPPINGS = {
"RunwareVideoInputsReferenceImagesMultipleImagesConnector": RunwareVideoInputsReferenceImagesMultipleImagesConnector,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"RunwareVideoInputsReferenceImagesMultipleImagesConnector": "Runware Video Inference Inputs Reference Images Multiple Images Connector",
}