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(