Fix FluxTrainAndValidateLoop.validate

2025-02-03 23:11:31 +03:00
parent f7025638fa
commit 74611324dc
1 changed files with 1 additions and 7 deletions
@@ -980,15 +980,9 @@ class FluxTrainAndValidateLoop:
        return (trainer, network_trainer.global_step)

    def validate(self, network_trainer, validation_settings=None):
-        params = (
-            network_trainer.accelerator, 
-            network_trainer.args, 
+        params = ( 
            network_trainer.current_epoch.value, 
            network_trainer.global_step,
-            network_trainer.unet,
-            network_trainer.vae,
-            network_trainer.text_encoder,
-            network_trainer.sample_prompts_te_outputs,
            validation_settings
        )
        network_trainer.optimizer_eval_fn()