Fix issue with vae file path validation

pull/2454/head
bmaltais 2024-05-06 06:43:43 -04:00
parent d26dad20b6
commit 8f8907f36c
6 changed files with 48 additions and 14 deletions

View File

@ -1 +1 @@
v24.1.0
v24.1.1

View File

@ -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.

View File

@ -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

View File

@ -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
#

View File

@ -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
#

View File

@ -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,