diff --git a/finetune/train_cogvideox_lora.py b/finetune/train_cogvideox_lora.py
index f2bb1c4..e12b3d5 100644
--- a/finetune/train_cogvideox_lora.py
+++ b/finetune/train_cogvideox_lora.py
@@ -1375,7 +1375,7 @@ def main(args):
                         num_frames=num_frames,
                         vae_scale_factor_spatial=vae_scale_factor_spatial,
                         patch_size=model_config.patch_size,
-                        patch_size_t=model_config.patch_size_t,
+                        patch_size_t=model_config.patch_size_t if model_config.patch_size_t is not None else 1,
                         attention_head_dim=model_config.attention_head_dim,
                         device=accelerator.device,
                     )