From 03866af5f994d963468bd2c87147cd0b3d20dd53 Mon Sep 17 00:00:00 2001 From: luanjintai Date: Mon, 10 Apr 2023 11:11:09 +0800 Subject: [PATCH] fix wrong parameter name for accelerate --- .../dreambooth_inpaint/train_dreambooth_inpaint_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py index 0522488f28..07df6f2011 100644 --- a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py +++ b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py @@ -411,7 +411,7 @@ def main(): mixed_precision=args.mixed_precision, log_with="tensorboard", logging_dir=logging_dir, - accelerator_project_config=accelerator_project_config, + project_config=accelerator_project_config, ) # Currently, it's not possible to do gradient accumulation when training two models with accelerate.accumulate