diff --git a/examples/dreambooth/train_dreambooth_lora_sdxl.py b/examples/dreambooth/train_dreambooth_lora_sdxl.py index e42311fb8f..85effa19b5 100644 --- a/examples/dreambooth/train_dreambooth_lora_sdxl.py +++ b/examples/dreambooth/train_dreambooth_lora_sdxl.py @@ -1764,7 +1764,7 @@ def main(args): accelerator, pipeline_args, epoch, - final_validation=True, + is_final_validation=True, ) if args.push_to_hub: