From 2cb9a1f29e5e86faca4d96ef685ff0669c8d9e93 Mon Sep 17 00:00:00 2001
From: hailin <hailin.zhao@gdzx.xyz>
Date: Fri, 6 Mar 2026 03:05:01 -0800
Subject: [PATCH] fix: add num_frames/height/width/fps/id columns to smoke test
 CSV

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 scripts/diffusion/gen_smoke_data.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/scripts/diffusion/gen_smoke_data.py b/scripts/diffusion/gen_smoke_data.py
index b107dea..b5e36b8 100644
--- a/scripts/diffusion/gen_smoke_data.py
+++ b/scripts/diffusion/gen_smoke_data.py
@@ -48,8 +48,13 @@ def main():
         torch.save(torch.randn(CLIP_DIM),                     clip_path)
 
         rows.append({
+            "id":             i,
             "path":           lat_path,   # 占位（cached 模式不读原始视频）
             "text":           f"smoke test sample {i}",
+            "num_frames":     1,          # 单帧图像
+            "height":         256,
+            "width":          256,
+            "fps":            24.0,
             "latents_path":   lat_path,
             "text_t5_path":   t5_path,
             "text_clip_path": clip_path,