Added support for multiple images as RefImages
This commit is contained in:
@@ -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",
|
||||
}
|
||||
Reference in New Issue
Block a user