1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

fix wrong parameter name for accelerate

This commit is contained in:
luanjintai
2023-04-10 11:11:09 +08:00
committed by Daniel Gu
parent a26f35ff2e
commit 03866af5f9

View File

@@ -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