From 1904947319da851024f71ee02ca4616ff4f91a2d Mon Sep 17 00:00:00 2001
From: henryfzh <59184714+henryfzh@users.noreply.github.com>
Date: Thu, 6 Nov 2025 16:47:34 -0800
Subject: [PATCH] Cleaning up coding

Updated comment in i2v_record_creator
---
 fastvideo/dataset/dataloader/record_schema.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/fastvideo/dataset/dataloader/record_schema.py b/fastvideo/dataset/dataloader/record_schema.py
index 632cb2d5d..ebb8cb5ea 100644
--- a/fastvideo/dataset/dataloader/record_schema.py
+++ b/fastvideo/dataset/dataloader/record_schema.py
@@ -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: