fix

Author: zhtmike

examples/wan2_2/wan/trainer/lora_trainer.py

@@ -141,6 +141,7 @@ def validate(self):
         logger.info("Running validation...")
         self.pipeline.model.set_train(False)
         self.pipeline.generate(**self.generation_config)
+        self.pipeline.model.set_train(True)
 
     def train(self, num_epochs):
         for epoch in range(num_epochs):