Fix small runtime issues

This commit is contained in:
JerryZhou54
2025-10-07 05:03:36 +00:00
parent aeb8f2e5ac
commit 55d8c1e5fb
3 changed files with 24 additions and 18 deletions
@@ -62,7 +62,7 @@ training_args=(
--max_train_steps 4000
--train_batch_size 1
--train_sp_batch_size 1
--gradient_accumulation_steps 2
--gradient_accumulation_steps 1
--num_latent_t 21
--num_height 448 # Updated to match Wan2.2 config
--num_width 832 # Updated to match Wan2.2 config