Add Sourceful provider Settings
This commit is contained in:
@@ -5,6 +5,7 @@ class imageInferenceInputs:
|
||||
"""Image Inference Inputs node for configuring image generation inputs"""
|
||||
|
||||
MAX_REFERENCE_IMAGES = 14
|
||||
MAX_SUPER_RESOLUTION_REFERENCE_IMAGES = 5
|
||||
|
||||
@classmethod
|
||||
def INPUT_TYPES(cls):
|
||||
@@ -27,6 +28,12 @@ class imageInferenceInputs:
|
||||
"default": "",
|
||||
})
|
||||
|
||||
for i in range(1, cls.MAX_SUPER_RESOLUTION_REFERENCE_IMAGES + 1):
|
||||
ordinal = rwUtils.getOrdinal(i)
|
||||
optionalInputs[f"Super Resolution Reference Image {i}"] = ("IMAGE", {
|
||||
"tooltip": f"Specifies {ordinal.capitalize()} Super Resolution Reference Image for the inputs.",
|
||||
})
|
||||
|
||||
return {
|
||||
"required": {},
|
||||
"optional": optionalInputs
|
||||
@@ -54,6 +61,10 @@ class imageInferenceInputs:
|
||||
if len(references) > 0:
|
||||
inputs["referenceImages"] = references
|
||||
|
||||
super_resolution_refs = self._collectSuperResolutionReferences(kwargs)
|
||||
if len(super_resolution_refs) > 0:
|
||||
inputs["superResolutionReferenceImages"] = super_resolution_refs
|
||||
|
||||
return (inputs,)
|
||||
|
||||
def _collectReferences(self, kwargs):
|
||||
@@ -91,3 +102,12 @@ class imageInferenceInputs:
|
||||
references.append(entry)
|
||||
|
||||
return references
|
||||
|
||||
def _collectSuperResolutionReferences(self, kwargs):
|
||||
"""Collect and convert super resolution reference images to list (same pattern as referenceImages)"""
|
||||
images = []
|
||||
for i in range(1, self.MAX_SUPER_RESOLUTION_REFERENCE_IMAGES + 1):
|
||||
image = kwargs.get(f"Super Resolution Reference Image {i}", None)
|
||||
if image is not None:
|
||||
images.append(rwUtils.convertTensor2IMG(image))
|
||||
return images
|
||||
|
||||
Reference in New Issue
Block a user