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
@@ -981,14 +981,8 @@ class FluxTrainAndValidateLoop:

    def validate(self, network_trainer, validation_settings=None):
        params = ( 
-            network_trainer.accelerator, 
-            network_trainer.args, 
            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()