From 46d0115c2b1552469b342497dd4fea18ce1f0aca Mon Sep 17 00:00:00 2001 From: Kadir Nar Date: Thu, 6 Apr 2023 12:55:43 +0300 Subject: [PATCH] The variable name has been updated. (#2970) --- .../dreambooth_inpaint/train_dreambooth_inpaint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py index c9b9211415b8..5158f9fc3bc0 100644 --- a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py +++ b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py @@ -405,14 +405,14 @@ def main(): args = parse_args() logging_dir = Path(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) + project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, mixed_precision=args.mixed_precision, log_with="tensorboard", logging_dir=logging_dir, - accelerator_project_config=accelerator_project_config, + project_config=project_config, ) # Currently, it's not possible to do gradient accumulation when training two models with accelerate.accumulate