From a113311e7762f4745d8cbce4c0bfaeecfe4ac617 Mon Sep 17 00:00:00 2001
From: Yongqi Chen <144848849+BrianChen1129@users.noreply.github.com>
Date: Wed, 30 Jul 2025 15:24:45 -0400
Subject: [PATCH] [Bugfix][Training]Fix Wan2.2 training vae config issue (#668)

---
 fastvideo/training/wan_training_pipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fastvideo/training/wan_training_pipeline.py b/fastvideo/training/wan_training_pipeline.py
index c9e7ab593..5af79519a 100644
--- a/fastvideo/training/wan_training_pipeline.py
+++ b/fastvideo/training/wan_training_pipeline.py
@@ -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(