@echo off


echo 1> F:\xampp\www\training.log

python -m venv xflux_env

call xflux_env\Scripts\deactivate.bat
call xflux_env\Scripts\activate.bat

rem set MODEL_NAME="F:/stable-g2/stable-diffusion-webui-forge/models/Stable-diffusion/flux1-schnell.safetensors"
rem set INSTANCE_DIR="analie1"
rem set OUTPUT_DIR="trained-flux"

rem default flux
accelerate launch "train_flux_lora_deepspeed.py" --config "train_configs/test_lora.yaml"

echo 0> F:\xampp\www\training.log


rem kohya flux
rem accelerate launch "train_dreambooth_flux.py" --pretrained_model_name_or_path="%MODEL_NAME%" --instance_data_dir="%INSTANCE_DIR%" --output_dir="%OUTPUT_DIR%" --mixed_precision="bf16" --instance_prompt="a photo of analie" --resolution=1024 --train_batch_size=1 --guidance_scale=1 --gradient_accumulation_steps=4 --optimizer="prodigy" --learning_rate=1. --report_to="wandb" --lr_scheduler="constant" --lr_warmup_steps=0 --max_train_steps=500 --validation_prompt="A photo of analie wearing a diaper" --validation_epochs=25 --seed="0"


rem kohya sd
rem accelerate launch --num_cpu_threads_per_process=8 "train_network.py" --enable_bucket --pretrained_model_name_or_path="G:/stable-g/stable-diffusion-webui-forge/models/Stable-diffusion/%in_pretrain%.safetensors" --train_data_dir="C:/trainimages/img" --resolution=1024,1024 --output_dir="C:/trainimages/model" --logging_dir="C:/trainimages/log" --network_alpha="128" --save_model_as=safetensors --network_module=networks.lora --text_encoder_lr=5e-5 --unet_lr=0.0001 --lr_scheduler_num_cycles="1" --learning_rate="0.00003" --lr_scheduler="cosine" --train_batch_size="1" --max_train_steps="%max_steps%" --save_every_n_epochs="1" --mixed_precision="fp16" --save_precision="fp16" --seed="0" --caption_extension=".txt" --cache_latents --max_data_loader_n_workers="1" --clip_skip=1 --optimizer_type="AdamW8bit" --bucket_reso_steps=64 --mem_eff_attn --gradient_checkpointing --xformers --bucket_no_upscale