support dpo training

2026-03-21 08:08:13 +00:00 · 2025-09-22 10:14:17 +08:00
parent b0abdaffb4
commit bf7b339efb
7 changed files with 213 additions and 6 deletions
--- a/examples/wanvideo/model_training/train.py
+++ b/examples/wanvideo/model_training/train.py
@@ -82,7 +82,7 @@ class WanTrainingModule(DiffusionTrainingModule):
        return {**inputs_shared, **inputs_posi}
    
    
-    def forward(self, data, inputs=None):
+    def forward(self, data, inputs=None, **kwargs):
        if inputs is None: inputs = self.forward_preprocess(data)
        models = {name: getattr(self.pipe, name) for name in self.pipe.in_iteration_models}
        loss = self.pipe.training_loss(**models, **inputs)