Skip to content

Commit c1f367d

Browse files
kadirnarw4ffl35
authored andcommitted
The variable name has been updated. (huggingface#2970)
1 parent 3200274 commit c1f367d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -405,14 +405,14 @@ def main():
405405
args = parse_args()
406406
logging_dir = Path(args.output_dir, args.logging_dir)
407407

408-
accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
408+
project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit)
409409

410410
accelerator = Accelerator(
411411
gradient_accumulation_steps=args.gradient_accumulation_steps,
412412
mixed_precision=args.mixed_precision,
413413
log_with="tensorboard",
414414
logging_dir=logging_dir,
415-
accelerator_project_config=accelerator_project_config,
415+
project_config=project_config,
416416
)
417417

418418
# Currently, it's not possible to do gradient accumulation when training two models with accelerate.accumulate

0 commit comments

Comments
 (0)