File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
examples/research_projects/dreambooth_inpaint Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -405,14 +405,14 @@ def main():
405
405
args = parse_args ()
406
406
logging_dir = Path (args .output_dir , args .logging_dir )
407
407
408
- accelerator_project_config = ProjectConfiguration (total_limit = args .checkpoints_total_limit )
408
+ project_config = ProjectConfiguration (total_limit = args .checkpoints_total_limit )
409
409
410
410
accelerator = Accelerator (
411
411
gradient_accumulation_steps = args .gradient_accumulation_steps ,
412
412
mixed_precision = args .mixed_precision ,
413
413
log_with = "tensorboard" ,
414
414
logging_dir = logging_dir ,
415
- accelerator_project_config = accelerator_project_config ,
415
+ project_config = project_config ,
416
416
)
417
417
418
418
# Currently, it's not possible to do gradient accumulation when training two models with accelerate.accumulate
You can’t perform that action at this time.
0 commit comments