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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user