Unverified Commit 0d4b459b authored by nickkolok's avatar nickkolok Committed by GitHub
Browse files

Update train_dreambooth.py - fix typos (#5539)

parent cee1cd6e
...@@ -1167,7 +1167,7 @@ def main(args): ...@@ -1167,7 +1167,7 @@ def main(args):
if args.resume_from_checkpoint != "latest": if args.resume_from_checkpoint != "latest":
path = os.path.basename(args.resume_from_checkpoint) path = os.path.basename(args.resume_from_checkpoint)
else: else:
# Get the mos recent checkpoint # Get the most recent checkpoint
dirs = os.listdir(args.output_dir) dirs = os.listdir(args.output_dir)
dirs = [d for d in dirs if d.startswith("checkpoint")] dirs = [d for d in dirs if d.startswith("checkpoint")]
dirs = sorted(dirs, key=lambda x: int(x.split("-")[1])) dirs = sorted(dirs, key=lambda x: int(x.split("-")[1]))
...@@ -1364,7 +1364,7 @@ def main(args): ...@@ -1364,7 +1364,7 @@ def main(args):
if global_step >= args.max_train_steps: if global_step >= args.max_train_steps:
break break
# Create the pipeline using using the trained modules and save it. # Create the pipeline using the trained modules and save it.
accelerator.wait_for_everyone() accelerator.wait_for_everyone()
if accelerator.is_main_process: if accelerator.is_main_process:
pipeline_args = {} pipeline_args = {}
......
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