[bugfix] [misc] Use training_state_checkpointing_steps in scripts/ (#846)

Author: SolitaryThinker

scripts/distill/v1_distill_dmd_wan.sh

@@ -32,7 +32,7 @@ torchrun --nnodes 1 --nproc_per_node $NUM_GPUS \
     --max_train_steps 30000 \
     --learning_rate 2e-6 \
     --mixed_precision "bf16" \
-    --checkpointing_steps 400 \
+    --training_state_checkpointing_steps 400 \
     --validation_steps 100 \
     --validation_sampling_steps "3" \
     --log_validation \

scripts/distill/v1_distill_dmd_wan_VSA.sh

@@ -33,7 +33,7 @@ torchrun --nnodes 1 --nproc_per_node $NUM_GPUS \
     --max_train_steps 30000 \
     --learning_rate 2e-6 \
     --mixed_precision "bf16" \
-    --checkpointing_steps 400 \
+    --training_state_checkpointing_steps 400 \
     --validation_steps 100 \
     --validation_sampling_steps "3" \
     --log_validation \

scripts/finetune/finetune_v1.sh

@@ -30,7 +30,7 @@ torchrun --nnodes 1 --nproc_per_node $NUM_GPUS\
     --max_train_steps=5000 \
     --learning_rate=1e-6\
     --mixed_precision="bf16"\
-    --checkpointing_steps=6000 \
+    --training_state_checkpointing_steps 6000 \
     --validation_steps 200\
     --validation_sampling_steps "2,4,8" \
     --log_validation \

scripts/finetune/finetune_v1_VSA.sh

@@ -36,7 +36,7 @@ torchrun --nnodes 1 --nproc_per_node $NUM_GPUS \
     --max_train_steps 30000 \
     --learning_rate 1e-6 \
     --mixed_precision "bf16" \
-    --checkpointing_steps 6000 \
+    --training_state_checkpointing_steps 6000 \
     --validation_steps 100 \
     --validation_sampling_steps "50" \
     --log_validation \