Removed skyreel conditioning

This commit is contained in:
Sirshendu Ganguly
2026-04-28 19:33:50 +05:30
parent c32e0448f2
commit a47bb0c873
3 changed files with 16 additions and 13 deletions
+14 -11
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 based on model requirements:
- skywork:skyreels@v4 -> grouped objects [{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 = isinstance(model, str) and model.strip().lower() == "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)}")
@@ -389,11 +393,10 @@ class txt2vid:
# Merge each input from inputs (only actual input data, not provider settings)
for key, value in inputs.items():
# Normalize referenceImages:
# - skyreels -> grouped objects
# - others -> flat string[] payload
# Normalize referenceImages by payload style:
# grouped objects when using ImagesN; otherwise flat string[]
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
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long