Updates

SC_finetuning_gui.py

@@ -60,7 +60,6 @@ def save_configuration(
     effnet_checkpoint_path,
     previewer_checkpoint_path,
     dataset_config_path,
-    sample_prompts_path,
     output_dir,
     logging_dir,
     flip_aug,
@@ -183,7 +182,6 @@ def open_configuration(
     effnet_checkpoint_path,
     previewer_checkpoint_path,
     dataset_config_path,
-    sample_prompts_path,
     output_dir,
     logging_dir,
     flip_aug,
@@ -313,7 +311,6 @@ def train_model(
     effnet_checkpoint_path,
     previewer_checkpoint_path,
     dataset_config_path,
-    sample_prompts_path,
     output_dir,
     logging_dir,
     flip_aug,
@@ -470,7 +467,6 @@ def train_model(
     run_cmd += f' --effnet_checkpoint_path "{effnet_checkpoint_path}"'
     run_cmd += f' --previewer_checkpoint_path "{previewer_checkpoint_path}"'
     run_cmd += f' --dataset_config "{dataset_config_path}"'
-    # run_cmd += f' --sample_prompts "{sample_prompts_path}"'
 
     run_cmd += run_cmd_advanced_training(
         adaptive_noise_scale=adaptive_noise_scale,
@@ -618,7 +614,7 @@ def sc_finetune_tab(headless=False):
                 )
                 effnet_checkpoint_path_file.click(
                     get_file_path,
-                    inputs=effnet_checkpoint_path,
+                    inputs=[effnet_checkpoint_path, gr.Textbox(value=".safetensors", visible=False, min_width=0)],
                     outputs=effnet_checkpoint_path,
                     show_progress=False,
                 )
@@ -632,13 +628,13 @@ def sc_finetune_tab(headless=False):
                 )
                 previewer_checkpoint_path_file.click(
                     get_file_path,
-                    inputs=previewer_checkpoint_path,
+                    inputs=[previewer_checkpoint_path, gr.Textbox(value=".safetensors", visible=False, min_width=0)],
                     outputs=previewer_checkpoint_path,
                     show_progress=False,
                 )
             with gr.Row():
                 dataset_config_path = gr.Textbox(
-                    label='Dataset toml path',
+                    label='Dataset toml file path',
                     placeholder='enter the path to the finetuning dataset toml file',
                 )
                 dataset_config_path_file = gr.Button(
@@ -647,24 +643,10 @@ def sc_finetune_tab(headless=False):
                 )
                 dataset_config_path_file.click(
                     get_file_path,
-                    inputs=dataset_config_path,
+                    inputs=[dataset_config_path, gr.Textbox(value=".toml", visible=False)],
                     outputs=dataset_config_path,
                     show_progress=False,
                 )
-                sample_prompts_path = gr.Textbox(
-                    label='Sample image prompt path',
-                    placeholder='enter the path to the sample image prompt file',
-                )
-                sample_prompts_path_file = gr.Button(
-                    document_symbol,
-                    elem_id='open_folder_small',
-                )
-                sample_prompts_path_file.click(
-                    get_file_path,
-                    inputs=sample_prompts_path,
-                    outputs=sample_prompts_path,
-                    show_progress=False,
-                )
             with gr.Row():
                 logging_dir = gr.Textbox(
                     label="Logging folder",
@@ -716,11 +698,6 @@ def sc_finetune_tab(headless=False):
             #     inputs=[dataset_config_path],
             #     outputs=[dataset_config_path],
             # )
-            # sample_prompts_path.change(
-            #     remove_doublequote,
-            #     inputs=[sample_prompts_path],
-            #     outputs=[sample_prompts_path],
-            # )
 
         with gr.Tab("Parameters"):
 
@@ -810,7 +787,6 @@ def list_presets(path):
             effnet_checkpoint_path,
             previewer_checkpoint_path,
             dataset_config_path,
-            sample_prompts_path,
             output_dir,
             logging_dir,
             advanced_training.flip_aug,

examples/stable_cascade/preset-13-images-1800steps-test.json

@@ -0,0 +1,83 @@
+{
+  "adaptive_noise_scale": 0,
+  "additional_parameters": "--max_grad_norm=0.0 --no_half_vae",
+  "block_lr": "",
+  "bucket_no_upscale": false,
+  "bucket_reso_steps": 64,
+  "cache_latents": true,
+  "cache_latents_to_disk": true,
+  "caption_dropout_every_n_epochs": 0.0,
+  "caption_dropout_rate": 0,
+  "caption_extension": "",
+  "clip_skip": "1",
+  "color_aug": false,
+  "dataset_config_path": "D:/kohya_ss/examples/stable_cascade/test_dataset.toml",
+  "dataset_repeats": "20",
+  "effnet_checkpoint_path": "E:/models/stable_cascade/effnet_encoder.safetensors",
+  "epoch": 500,
+  "flip_aug": false,
+  "full_bf16": true,
+  "full_fp16": false,
+  "gpu_ids": "",
+  "gradient_accumulation_steps": 1.0,
+  "gradient_checkpointing": true,
+  "keep_tokens": "0",
+  "learning_rate": 1e-05,
+  "learning_rate_te": 1e-05,
+  "learning_rate_te1": 1e-05,
+  "learning_rate_te2": 1e-05,
+  "logging_dir": "",
+  "lr_scheduler": "constant",
+  "lr_scheduler_args": "",
+  "lr_warmup": 10,
+  "max_data_loader_n_workers": "0",
+  "max_timestep": 1000,
+  "max_token_length": "75",
+  "max_train_epochs": "60",
+  "mem_eff_attn": false,
+  "min_snr_gamma": 0,
+  "min_timestep": 0,
+  "mixed_precision": "bf16",
+  "model_list": "custom",
+  "multi_gpu": false,
+  "multires_noise_discount": 0,
+  "multires_noise_iterations": 0,
+  "noise_offset": 0,
+  "noise_offset_type": "Original",
+  "num_cpu_threads_per_process": 4,
+  "num_machines": 1,
+  "num_processes": 1,
+  "optimizer": "Adafactor",
+  "optimizer_args": "scale_parameter=False relative_step=False warmup_init=False weight_decay=0.01",
+  "output_dir": "E:/model/test",
+  "output_name": "ted-test-no-te",
+  "persistent_data_loader_workers": false,
+  "pretrained_model_name_or_path": "E:/models/stable_cascade/stage_c_bf16.safetensors",
+  "previewer_checkpoint_path": "E:/models/stable_cascade/previewer.safetensors",
+  "random_crop": false,
+  "resume": "",
+  "sample_every_n_epochs": 0,
+  "sample_every_n_steps": 100,
+  "sample_prompts": "a zxc on the surface of the moon holding an orange.  --w 1024 --h 1024 --l 4 --s 30",
+  "sample_prompts_path": "D:/kohya_ss/examples/stable_cascade/prompt.txt",
+  "sample_sampler": "dpm_2",
+  "save_every_n_epochs": 0,
+  "save_every_n_steps": 100,
+  "save_last_n_steps": 0,
+  "save_last_n_steps_state": 0,
+  "save_model_as": "safetensors",
+  "save_precision": "bf16",
+  "save_state": false,
+  "scale_v_pred_loss_like_noise_pred": false,
+  "seed": "",
+  "shuffle_caption": false,
+  "train_batch_size": 3,
+  "train_text_encoder": true,
+  "use_wandb": false,
+  "v2": false,
+  "v_parameterization": false,
+  "v_pred_like_loss": 0,
+  "vae_batch_size": 0,
+  "wandb_api_key": "",
+  "xformers": "none"
+}

examples/stable_cascade/test_dataset/1.txt

@@ -0,0 +1 @@
+zxc looking to the left

examples/stable_cascade/test_dataset/10.txt

@@ -0,0 +1 @@
+a photo of zxc

examples/stable_cascade/test_dataset/11.txt

@@ -0,0 +1 @@
+zxc writing on a piece of paper

examples/stable_cascade/test_dataset/2.txt

@@ -0,0 +1 @@
+zxc sitting on a couch

examples/stable_cascade/test_dataset/3.txt

@@ -0,0 +1 @@
+zxc standing

examples/stable_cascade/test_dataset/4.txt

@@ -0,0 +1 @@
+zxc standing rear view

examples/stable_cascade/test_dataset/5.txt

@@ -0,0 +1 @@
+zxc sitting

examples/stable_cascade/test_dataset/8.txt

@@ -0,0 +1 @@
+zxc standing amongst sausages