Added backward compatibility

This commit is contained in:
Sirshendu Ganguly
2026-04-28 19:09:17 +05:30
parent acc3ac6456
commit fbdb3f15bb
4 changed files with 53 additions and 31 deletions
+12 -8
View File
@@ -5,11 +5,12 @@ import comfy.model_management
class txt2vid:
@staticmethod
def _format_reference_images(referenceImages, model):
def _format_reference_images(referenceImages):
"""
Shape referenceImages per model requirements:
- skywork:skyreels@v4 -> list of groups: {type, tag, images, audio?}
- others -> flat list[str] (legacy behavior)
Shape referenceImages by payload style:
- If any entry uses `images` (Images1..N style), return grouped objects
[{type, tag, images, audio?}]
- Otherwise return flat list[str] (legacy behavior)
"""
if referenceImages is None:
return []
@@ -17,9 +18,12 @@ class txt2vid:
if not isinstance(referenceImages, list):
referenceImages = [referenceImages]
is_skyreels = model == "skywork:skyreels@v4"
has_grouped_images = any(
isinstance(ref, dict) and isinstance(ref.get("images"), list) and len(ref.get("images")) > 0
for ref in referenceImages
)
if is_skyreels:
if has_grouped_images:
grouped = []
for idx, ref in enumerate(referenceImages, start=1):
images = []
@@ -357,7 +361,7 @@ class txt2vid:
print(f"[Debugging] Frame images array: {rwUtils.sanitize_for_logging(frameImages)}")
# Add referenceImages if provided from Runware Reference Images node
if referenceImages is not None and len(referenceImages) > 0:
formattedReferenceImages = self._format_reference_images(referenceImages, model)
formattedReferenceImages = self._format_reference_images(referenceImages)
if len(formattedReferenceImages) > 0:
genConfig[0]["referenceImages"] = formattedReferenceImages
print(f"[Debugging] Reference images array: {rwUtils.sanitize_for_logging(formattedReferenceImages)}")
@@ -391,7 +395,7 @@ class txt2vid:
for key, value in inputs.items():
# Normalize referenceImages so non-SkyReels models keep legacy string[] payload
if key == "referenceImages":
formatted_input_refs = self._format_reference_images(value, model)
formatted_input_refs = self._format_reference_images(value)
if len(formatted_input_refs) > 0:
genConfig[0]["inputs"][key] = formatted_input_refs
continue