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: