From 81d9a873627ec0b5a18c091f63fb8a0fcf6d2f5c Mon Sep 17 00:00:00 2001 From: nickkolok Date: Thu, 26 Oct 2023 14:17:46 +0300 Subject: [PATCH] Update train_dreambooth.py - fix typos --- examples/dreambooth/train_dreambooth.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 6ad79a47deb5..6d59ee4de383 100644 --- a/examples/dreambooth/train_dreambooth.py +++ b/examples/dreambooth/train_dreambooth.py @@ -1167,7 +1167,7 @@ def compute_text_embeddings(prompt): if args.resume_from_checkpoint != "latest": path = os.path.basename(args.resume_from_checkpoint) else: - # Get the mos recent checkpoint + # Get the most recent checkpoint dirs = os.listdir(args.output_dir) dirs = [d for d in dirs if d.startswith("checkpoint")] dirs = sorted(dirs, key=lambda x: int(x.split("-")[1])) @@ -1364,7 +1364,7 @@ def compute_text_embeddings(prompt): if global_step >= args.max_train_steps: 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() if accelerator.is_main_process: pipeline_args = {}