test timestep fix

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

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)