update readme for vae training

jy0205 · Nov 29, 2024 · 8acc528 · 8acc528
1 parent 1f864c4
commit 8acc528
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train/train_video_vae.py b/train/train_video_vae.py
@@ -223,6 +223,7 @@ def main(args):
         world_size=world_size,
         rank=global_rank,
         image_mix_ratio=args.image_mix_ratio,
+        use_image_video_mixed_training=args.use_image_video_mixed_training,
     )
 
     torch.distributed.barrier()