update

Dockerfile

@@ -1,3 +1,6 @@
+
+# pytorch/pytorch:2.3.0-cuda12.1-cudnn8-devel
+
 # syntax=docker/dockerfile:1
 ARG UID=1000
 ARG VERSION=EDGE

run-train-extd.cmd

@@ -1,31 +0,0 @@
-$env:CUDA_VISIBLE_DEVICES = "1"
-
-accelerate launch --num_cpu_threads_per_process 1 --mixed_precision bf16 ^
-  sdxl_train_network.py ^
-  --pretrained_model_name_or_path="./models/stable-diffusion-xl-base-1.0" ^
-  --train_data_dir="./train_data" ^
-  --output_dir="./output_model" ^
-  --logging_dir="./logs" ^
-  --output_name="karina" ^
-  --network_module=networks.lora ^
-  --network_dim=32 ^
-  --network_alpha=16 ^
-  --learning_rate=1e-4 ^
-  --optimizer_type="AdamW8bit" ^
-  --lr_scheduler="cosine" ^
-  --lr_warmup_steps=100 ^
-  --max_train_epochs=15 ^
-  --save_every_n_epochs=1 ^
-  --mixed_precision="bf16" ^
-  --save_precision="bf16" ^
-  --cache_latents ^
-  --cache_latents_to_disk ^
-  --cache_text_encoder_outputs ^
-  --gradient_checkpointing ^
-  --xformers ^
-  --seed=42 ^
-  --bucket_no_upscale ^
-  --min_bucket_reso=512 ^
-  --max_bucket_reso=2048 ^
-  --bucket_reso_steps=64 ^
-  --resolution="1024,1024"

run-train-simple.cmd

@@ -0,0 +1 @@
+accelerate launch --num_cpu_threads_per_process 8 train_network.py --config_file=config_5080.json

run-train.cmd

@@ -1 +1,31 @@
-accelerate launch --num_cpu_threads_per_process 8 train_network.py --config_file=config_5080.json
+setx CUDA_VISIBLE_DEVICES "1"
+
+accelerate launch --num_cpu_threads_per_process 1 --mixed_precision bf16 ^
+  sdxl_train_network.py ^
+  --pretrained_model_name_or_path="./models/stable-diffusion-xl-base-1.0" ^
+  --train_data_dir="./dataset/mainchar" ^
+  --output_dir="./output_model" ^
+  --logging_dir="./logs" ^
+  --output_name="karina" ^
+  --network_module=networks.lora ^
+  --network_dim=32 ^
+  --network_alpha=16 ^
+  --learning_rate=1e-4 ^
+  --optimizer_type="AdamW8bit" ^
+  --lr_scheduler="cosine" ^
+  --lr_warmup_steps=100 ^
+  --max_train_epochs=15 ^
+  --save_every_n_epochs=1 ^
+  --mixed_precision="bf16" ^
+  --save_precision="bf16" ^
+  --cache_latents ^
+  --cache_latents_to_disk ^
+  --cache_text_encoder_outputs ^
+  --gradient_checkpointing ^
+  --xformers ^
+  --seed=42 ^
+  --bucket_no_upscale ^
+  --min_bucket_reso=512 ^
+  --max_bucket_reso=2048 ^
+  --bucket_reso_steps=64 ^
+  --resolution="1024,1024"