nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Bugfix] Add i2v vae loading (#686)

Browse Source
This commit is contained in:
Yongqi Chen
2025-08-01 23:15:37 -04:00
committed by GitHub
parent 31200fbb83
commit dfe0b288e1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastvideo/training/wan_i2v_training_pipeline.py
+1 -1
View File
@@ -27,7 +27,7 @@ class WanI2VTrainingPipeline(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(