[refactor]: update train scripts to use generic runner script

Author: techverve

scripts/distill/v1_distill_dmd_wan.sh

@@ -10,7 +10,9 @@ export TOKENIZERS_PARALLELISM=false
 
 # make sure that num_latent_t is a multiple of sp_size
 torchrun --nnodes 1 --nproc_per_node $NUM_GPUS \
-    fastvideo/training/wan_distillation_pipeline.py \
+    fastvideo/training/runner.py \
+    --pipeline_class WanDistillationPipeline \
+    --pipeline_module fastvideo.training.wan_distillation_pipeline \
     --model_path Wan-AI/Wan2.1-T2V-1.3B-Diffusers \
     --real_score_model_path Wan-AI/Wan2.1-T2V-1.3B-Diffusers \
     --fake_score_model_path Wan-AI/Wan2.1-T2V-1.3B-Diffusers \

scripts/distill/v1_distill_dmd_wan_VSA.sh

@@ -11,7 +11,9 @@ export TOKENIZERS_PARALLELISM=false
 # Train generator with VSA
 # Make sure that num_latent_t is a multiple of sp_size
 torchrun --nnodes 1 --nproc_per_node $NUM_GPUS \
-    fastvideo/training/wan_distillation_pipeline.py \
+    fastvideo/training/runner.py \
+    --pipeline_class WanDistillationPipeline \
+    --pipeline_module fastvideo.training.wan_distillation_pipeline \
     --model_path Wan-AI/Wan2.1-T2V-1.3B-Diffusers \
     --real_score_model_path Wan-AI/Wan2.1-T2V-1.3B-Diffusers \
     --fake_score_model_path Wan-AI/Wan2.1-T2V-1.3B-Diffusers \

scripts/finetune/finetune_v1.sh

@@ -11,7 +11,9 @@ NUM_GPUS=4
 
 # Make sure that num_latent_t is a multiple of sp_size
 torchrun --nnodes 1 --nproc_per_node $NUM_GPUS\
-    fastvideo/training/wan_training_pipeline.py\
+    fastvideo/training/runner.py\
+    --pipeline_class WanTrainingPipeline \
+    --pipeline_module fastvideo.training.wan_training_pipeline \
     --model_path Wan-AI/Wan2.1-T2V-1.3B-Diffusers \
     --inference_mode False\
     --pretrained_model_name_or_path Wan-AI/Wan2.1-T2V-1.3B-Diffusers \