[Bugfix][Training]Fix Wan2.2 training vae config issue (#668)

2025-07-30 12:24:45 -07:00
parent 44da97da92
commit a113311e77
1 changed files with 1 additions and 1 deletions
@@ -19,7 +19,7 @@ class WanTrainingPipeline(TrainingPipeline):
    """
    A training pipeline for Wan.
    """
-    _required_config_modules = ["scheduler", "transformer"]
+    _required_config_modules = ["scheduler", "transformer", "vae"]

    def initialize_pipeline(self, fastvideo_args: FastVideoArgs):
        self.modules["scheduler"] = FlowUniPCMultistepScheduler(