merged from main

Author: Serenagu525

src/maxdiffusion/configs/ltx_video.yml

@@ -83,10 +83,9 @@ ici_data_parallelism: 1
 ici_fsdp_parallelism: -1  # recommended ICI axis to be auto-sharded
 ici_tensor_parallelism: 1
 
-
-
+allow_split_physical_axes: False
 learning_rate_schedule_steps: -1
-max_train_steps: 500 #TODO: change this
+max_train_steps: 500
 pretrained_model_name_or_path: ''
 unet_checkpoint: ''
 dataset_name: 'diffusers/pokemon-gpt4-captions'