From acd65e42e4d92169c2d992c607d95d16fd5c98e1 Mon Sep 17 00:00:00 2001
From: Hangliang Ding <29483524+foreverpiano@users.noreply.github.com>
Date: Fri, 7 Feb 2025 01:15:57 +0800
Subject: [PATCH] Update num_frames

---
 scripts/distill/distill_hunyuan_dmd.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/distill/distill_hunyuan_dmd.sh b/scripts/distill/distill_hunyuan_dmd.sh
index 20cf2a921..996a65efd 100644
--- a/scripts/distill/distill_hunyuan_dmd.sh
+++ b/scripts/distill/distill_hunyuan_dmd.sh
@@ -36,7 +36,7 @@ torchrun --nnodes 1 --nproc_per_node 4\
     --tracker_project_name Hunyuan_Distill \
     --num_height 720 \
     --num_width 1280 \
-    --num_frames 29 \
+    --num_frames 125 \
     --shift 17 \
     --validation_guidance_scale "1.0" \
     --num_euler_timesteps 50 \
@@ -45,4 +45,4 @@ torchrun --nnodes 1 --nproc_per_node 4\
     --master_weight_type "bf16" \
     --optimizer "AdamW" \
     --use_8bit_adam \
-    --generator_update_steps 5
\ No newline at end of file
+    --generator_update_steps 3