From ee736ecb481a1e0b9e2e01a6299e0c396c2afd83 Mon Sep 17 00:00:00 2001 From: RandNMR73 Date: Thu, 2 Oct 2025 05:23:45 +0000 Subject: [PATCH] fix required configs in distillation pipelines --- fastvideo/training/wan_distillation_pipeline.py | 5 +---- fastvideo/training/wan_i2v_distillation_pipeline.py | 5 +---- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/fastvideo/training/wan_distillation_pipeline.py b/fastvideo/training/wan_distillation_pipeline.py index 246d3f0c5..cf8e7d899 100644 --- a/fastvideo/training/wan_distillation_pipeline.py +++ b/fastvideo/training/wan_distillation_pipeline.py @@ -20,10 +20,7 @@ class WanDistillationPipeline(DistillationPipeline): A distillation pipeline for Wan that uses a single transformer model. The main transformer serves as the student model, and copies are made for teacher and critic. """ - _required_config_modules = [ - "scheduler", "transformer", "vae", "real_score_transformer", - "fake_score_transformer" - ] + _required_config_modules = ["scheduler", "transformer", "vae"] def initialize_pipeline(self, fastvideo_args: FastVideoArgs): """Initialize Wan-specific scheduler.""" diff --git a/fastvideo/training/wan_i2v_distillation_pipeline.py b/fastvideo/training/wan_i2v_distillation_pipeline.py index 47b5d73de..9c5fee67a 100644 --- a/fastvideo/training/wan_i2v_distillation_pipeline.py +++ b/fastvideo/training/wan_i2v_distillation_pipeline.py @@ -29,10 +29,7 @@ class WanI2VDistillationPipeline(DistillationPipeline): A distillation pipeline for Wan that uses a single transformer model. The main transformer serves as the student model, and copies are made for teacher and critic. """ - _required_config_modules = [ - "scheduler", "transformer", "vae", "real_score_transformer", - "fake_score_transformer" - ] + _required_config_modules = ["scheduler", "transformer", "vae"] def initialize_pipeline(self, fastvideo_args: FastVideoArgs): """Initialize Wan-specific scheduler."""