From 3b771f51ec782a15939bfeda7ed651094183d05a Mon Sep 17 00:00:00 2001 From: bmaltais Date: Sat, 11 May 2024 09:10:39 -0400 Subject: [PATCH] Fix finetuning meta file creation (#2488) --- kohya_gui/finetune_gui.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/kohya_gui/finetune_gui.py b/kohya_gui/finetune_gui.py index 1e958b18e..c84922c0e 100644 --- a/kohya_gui/finetune_gui.py +++ b/kohya_gui/finetune_gui.py @@ -628,8 +628,8 @@ def train_model( run_cmd.append(caption_extension) # Add paths for the image folder and the caption metadata file - run_cmd.append(image_folder) - run_cmd.append(os.path.join(train_dir, caption_metadata_filename)) + run_cmd.append(rf"{image_folder}") + run_cmd.append(rf"{os.path.join(train_dir, caption_metadata_filename)}") # Include the full path flag if specified if full_path: @@ -641,16 +641,20 @@ def train_model( # Prepare environment variables env = setup_environment() + # Execute the command if not just for printing + if not print_only: + subprocess.run(run_cmd, env=env) + # create images buckets if generate_image_buckets: # Build the command to run the preparation script run_cmd = [ PYTHON, rf"{scriptdir}/sd-scripts/finetune/prepare_buckets_latents.py", - image_folder, - os.path.join(train_dir, caption_metadata_filename), - os.path.join(train_dir, latent_metadata_filename), - pretrained_model_name_or_path, + rf"{image_folder}", + rf"{os.path.join(train_dir, caption_metadata_filename)}", + rf"{os.path.join(train_dir, latent_metadata_filename)}", + rf"{pretrained_model_name_or_path}", "--batch_size", str(batch_size), "--max_resolution",