Skip to content

Commit 4c404e5

Browse files
authored
Fix unexpected keyword argument 'logging_dir' error.
1 parent 2c29830 commit 4c404e5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/dreambooth/train_inpainting_dreambooth.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,7 @@ def parse_args(input_args=None):
222222
help="The name of the repository to keep in sync with the local `output_dir`.",
223223
)
224224
parser.add_argument(
225-
"--logging_dir",
225+
"--project_dir",
226226
type=str,
227227
default="logs",
228228
help=(
@@ -428,13 +428,13 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
428428

429429

430430
def main(args):
431-
logging_dir = Path(args.output_dir, "0", args.logging_dir)
431+
project_dir = Path(args.output_dir, "0", args.project_dir)
432432

433433
accelerator = Accelerator(
434434
gradient_accumulation_steps=args.gradient_accumulation_steps,
435435
mixed_precision=args.mixed_precision,
436436
log_with="tensorboard",
437-
logging_dir=logging_dir,
437+
project_dir=project_dir,
438438
)
439439

440440
# Currently, it's not possible to do gradient accumulation when training two models with accelerate.accumulate

0 commit comments

Comments
 (0)