Skip to content

GitLab

Commit b5d0a913 authored by luanjintai's avatar luanjintai Committed by Will Berman
Browse files

fix wrong parameter name for accelerate

parent 983a7fbf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py
View file @ b5d0a913
...@@ -411,7 +411,7 @@ def main(): ...@@ -411,7 +411,7 @@ def main():
mixed_precision=args.mixed_precision, mixed_precision=args.mixed_precision,
log_with="tensorboard", log_with="tensorboard",
logging_dir=logging_dir, 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 # Currently, it's not possible to do gradient accumulation when training two models with accelerate.accumulate
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment