Cleaning up coding

Updated comment in i2v_record_creator
2026-01-17 11:52:54 -08:00
parent 5f8fb3cf5f
commit 1904947319
1 changed files with 1 additions and 2 deletions
@@ -72,8 +72,7 @@ def i2v_record_creator(batch: PreprocessBatch) -> list[dict[str, Any]]:
    """Create a record for the Parquet dataset with CLIP features."""
    records = basic_t2v_record_creator(batch)

-    # assert len(
-    #     batch.image_embeds) == 1, "image embedding should be a single tensor"
+    # Adapt for model doesn't have image decoder, e.g., Hunyuan
    if len(batch.image_embeds) == 0:
        image_embeds = None
    elif len(batch.image_embeds) == 1: