Merge pull request #130 from Mikhael-Danilov/patch-1

Fix FluxTrainAndValidateLoop.validate
2025-02-03 23:27:28 +02:00
parent f7025638fa 74611324dc
commit 639b3e80ba
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()