From 9ee229bb68e9d9fe334a48b192a80dc0fd7ee7bf Mon Sep 17 00:00:00 2001 From: RandNMR73 Date: Thu, 2 Oct 2025 09:57:56 +0000 Subject: [PATCH] test timestep fix --- examples/distill/SFWan2.2-A14B/distill_dmd.sh | 2 +- fastvideo/training/training_pipeline.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/distill/SFWan2.2-A14B/distill_dmd.sh b/examples/distill/SFWan2.2-A14B/distill_dmd.sh index 69f6389c1..c9bbfd882 100644 --- a/examples/distill/SFWan2.2-A14B/distill_dmd.sh +++ b/examples/distill/SFWan2.2-A14B/distill_dmd.sh @@ -80,7 +80,7 @@ parallel_args=( ) model_args=( - --model_path $GENERATOR_MODEL_PATH # TODO: check if you can remove this in this script + --model_path $GENERATOR_MODEL_PATH --pretrained_model_name_or_path $GENERATOR_MODEL_PATH --real_score_model_path $REAL_SCORE_MODEL_PATH --fake_score_model_path $FAKE_SCORE_MODEL_PATH diff --git a/fastvideo/training/training_pipeline.py b/fastvideo/training/training_pipeline.py index b50294ea8..c1af4416a 100644 --- a/fastvideo/training/training_pipeline.py +++ b/fastvideo/training/training_pipeline.py @@ -351,8 +351,8 @@ class TrainingPipeline(LoRAPipeline, ABC): if self.train_transformer_2: u = (1 - boundary_ratio ) + u * boundary_ratio # min: 1 - boundary_ratio, max: 1 - else: - u = u * (1 - boundary_ratio) # min: 0, max: 1 - boundary_ratio + # else: + # u = u * (1 - boundary_ratio) # min: 0, max: 1 - boundary_ratio indices = (u * self.noise_scheduler.config.num_train_timesteps).long() return self.noise_scheduler.timesteps[indices].to(device=device)