diff --git a/examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py b/examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py index aba9020f58..321b94bc6c 100644 --- a/examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py +++ b/examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py @@ -328,7 +328,7 @@ def main(): mixed_precision=args.mixed_precision, log_with=args.report_to, logging_dir=logging_dir, - accelerator_project_config=accelerator_project_config, + project_config=accelerator_project_config, ) # Make one log on every process with the configuration for debugging.