Update training script

kohya_diffusers_train.ps1

@@ -5,7 +5,7 @@ $pretrained_model_name_or_path = "D:\models\v1-5-pruned-mse-vae.ckpt"
 $train_dir = "D:\dreambooth\train_lego"
 $training_folder = "raw"
 
-$learning_rate = 5e-6
+$learning_rate = 1e-6
 $dataset_repeats = 40
 $train_batch_size = 8
 $epoch = 4
@@ -13,7 +13,7 @@ $save_every_n_epochs=1
 $mixed_precision="bf16"
 $num_cpu_threads_per_process=6
 
-$max_resolution = "512,512"
+$max_resolution = "576,576"
 
 # You should not have to change values past this point
 
@@ -28,7 +28,7 @@ $ErrorActionPreference = "Stop"
 $caption="lego set"
 
 $files = Get-ChildItem $train_dir\$training_folder\"*.*" -Include *.png,*.jpg,*.webp
-foreach ($file in $files) {New-Item -ItemType file -Path $train_dir\$training_folder -Name "$($file.BaseName).txt" -Value $caption}
+foreach ($file in $files) {New-Item -ItemType file -Path $train_dir\$training_folder -Name "$($file.BaseName).txt" -Value $caption -Force}
 
 
 # create caption json file
@@ -50,6 +50,7 @@ $repeats = $image_num * $dataset_repeats
 # calculate max_train_set
 $max_train_set = [Math]::Ceiling($repeats / $train_batch_size * $epoch)
 
+
 accelerate launch --num_cpu_threads_per_process $num_cpu_threads_per_process D:\kohya_ss\diffusers_fine_tuning\fine_tune.py `
     --pretrained_model_name_or_path=$pretrained_model_name_or_path `
     --in_json $train_dir"\meta_lat.json" `