diff --git a/.release b/.release index 959a88019..485bf12fd 100644 --- a/.release +++ b/.release @@ -1 +1 @@ -v24.1.0 \ No newline at end of file +v24.1.1 \ No newline at end of file diff --git a/README.md b/README.md index f8216c188..0c0cdb091 100644 --- a/README.md +++ b/README.md @@ -47,6 +47,7 @@ The GUI allows you to set the training parameters and generate and run the requi - [SDXL training](#sdxl-training) - [Masked loss](#masked-loss) - [Change History](#change-history) + - [v24.1.1](#v2411) - [v24.1.0](#v2410) ## 🦒 Colab @@ -441,6 +442,10 @@ ControlNet dataset is used to specify the mask. The mask images should be the RG ## Change History +### v24.1.1 + +- Fix small issue with VAE file path validation + ### v24.1.0 - To ensure cross-platform compatibility and security, the GUI now defaults to using "shell=False" when running subprocesses. This is based on documentation and should not cause issues on most platforms. However, some users have reported issues on specific platforms such as runpod and colab. PLease open an issue if you encounter any issues. diff --git a/kohya_gui/dreambooth_gui.py b/kohya_gui/dreambooth_gui.py index 113a7228e..c1b2283d2 100644 --- a/kohya_gui/dreambooth_gui.py +++ b/kohya_gui/dreambooth_gui.py @@ -541,7 +541,7 @@ def train_model( if not validate_folder_path(train_data_dir): return TRAIN_BUTTON_VISIBLE - if not validate_folder_path(vae): + if not validate_file_path(vae): return TRAIN_BUTTON_VISIBLE # # End of path validation diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py index 58411ddb2..50e30acb2 100644 --- a/kohya_gui/lora_gui.py +++ b/kohya_gui/lora_gui.py @@ -732,7 +732,7 @@ def train_model( if not validate_folder_path(train_data_dir): return TRAIN_BUTTON_VISIBLE - if not validate_folder_path(vae): + if not validate_file_path(vae): return TRAIN_BUTTON_VISIBLE # diff --git a/kohya_gui/textual_inversion_gui.py b/kohya_gui/textual_inversion_gui.py index b6fc72a7c..81bb36b12 100644 --- a/kohya_gui/textual_inversion_gui.py +++ b/kohya_gui/textual_inversion_gui.py @@ -542,7 +542,7 @@ def train_model( if not validate_folder_path(train_data_dir): return TRAIN_BUTTON_VISIBLE - if not validate_folder_path(vae): + if not validate_file_path(vae): return TRAIN_BUTTON_VISIBLE # diff --git a/test/config/finetune-AdamW-toml.json b/test/config/finetune-AdamW-toml.json index 9a818ebbc..5b396a70a 100644 --- a/test/config/finetune-AdamW-toml.json +++ b/test/config/finetune-AdamW-toml.json @@ -1,13 +1,14 @@ { "adaptive_noise_scale": 0, "additional_parameters": "", + "async_upload": false, "batch_size": "8", "block_lr": "", "bucket_no_upscale": false, "bucket_reso_steps": 1, "cache_latents": true, "cache_latents_to_disk": false, - "caption_dropout_every_n_epochs": 0.0, + "caption_dropout_every_n_epochs": 0, "caption_dropout_rate": 0, "caption_extension": ".txt", "caption_metadata_filename": "meta-1_cap.json", @@ -15,37 +16,61 @@ "color_aug": false, "create_buckets": false, "create_caption": true, - "dataset_config": "D:/kohya_ss/test/config/dataset-finetune.toml", + "dataset_config": "./test/config/dataset-finetune.toml", "dataset_repeats": "50", + "debiased_estimation_loss": false, + "dynamo_backend": "no", + "dynamo_mode": "default", + "dynamo_use_dynamic": false, + "dynamo_use_fullgraph": false, "epoch": 2, + "extra_accelerate_launch_args": "", "flip_aug": false, "full_bf16": false, "full_fp16": false, "full_path": true, "gpu_ids": "", - "gradient_accumulation_steps": 1.0, + "gradient_accumulation_steps": 1, "gradient_checkpointing": false, + "huber_c": 0.1, + "huber_schedule": "snr", + "huggingface_path_in_repo": "", + "huggingface_repo_id": "", + "huggingface_repo_type": "", + "huggingface_repo_visibility": "", + "huggingface_token": "", "image_folder": "", + "ip_noise_gamma": 0, + "ip_noise_gamma_random_strength": false, "keep_tokens": 0, "latent_metadata_filename": "meta-1_lat.json", "learning_rate": 1e-05, "learning_rate_te": 5e-06, "learning_rate_te1": 5e-06, - "learning_rate_te2": 0.0, + "learning_rate_te2": 0, "log_tracker_config": "", "log_tracker_name": "", + "log_with": "", "logging_dir": "./test/ft", + "loss_type": "l2", "lr_scheduler": "cosine_with_restarts", "lr_scheduler_args": "", "lr_warmup": 10, + "main_process_port": 0, + "masked_loss": false, "max_bucket_reso": "1024", - "max_data_loader_n_workers": "0", + "max_data_loader_n_workers": 0, "max_resolution": "512,512", "max_timestep": 1000, - "max_token_length": "75", - "max_train_epochs": "", - "max_train_steps": "20", + "max_token_length": 75, + "max_train_epochs": 0, + "max_train_steps": 20, "mem_eff_attn": false, + "metadata_author": "", + "metadata_description": "", + "metadata_license": "", + "metadata_tags": "", + "metadata_title": "", "min_bucket_reso": "256", "min_snr_gamma": 0, "min_timestep": 0, @@ -55,6 +80,7 @@ "multires_noise_discount": 0, "multires_noise_iterations": 0, "noise_offset": 0, + "noise_offset_random_strength": false, "noise_offset_type": "Original", "num_cpu_threads_per_process": 2, "num_machines": 1, @@ -67,10 +93,12 @@ "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5", "random_crop": false, "resume": "", + "resume_from_huggingface": "", "sample_every_n_epochs": 0, "sample_every_n_steps": 0, "sample_prompts": "", "sample_sampler": "euler_a", + "save_as_bool": false, "save_every_n_epochs": 1, "save_every_n_steps": 0, "save_last_n_steps": 0, @@ -78,17 +106,18 @@ "save_model_as": "safetensors", "save_precision": "bf16", "save_state": false, + "save_state_on_train_end": false, + "save_state_to_huggingface": false, "scale_v_pred_loss_like_noise_pred": false, "sdxl_cache_text_encoder_outputs": false, "sdxl_checkbox": false, "sdxl_no_half_vae": false, - "seed": "1234", + "seed": 1234, "shuffle_caption": false, "train_batch_size": 4, "train_dir": "./test", "train_text_encoder": true, "use_latent_files": "No", - "use_wandb": false, "v2": false, "v_parameterization": false, "v_pred_like_loss": 0,