| export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH | |
| export MODEL_NAME="/home/systemy/NAI" | |
| export OUTPUT_DIR="/mnt/d/jigsaw" | |
| accelerate launch train_dreambooth.py \ | |
| --pretrained_model_name_or_path="$MODEL_NAME" \ | |
| --pretrained_vae_name_or_path="stabilityai/sd-vae-ft-mse" \ | |
| --output_dir="/mnt/d/acuteoutput" \ | |
| --seed=3434554 \ | |
| --resolution=512 \ | |
| --train_batch_size=1 \ | |
| --train_text_encoder \ | |
| --mixed_precision="fp16" \ | |
| --use_8bit_adam \ | |
| --gradient_accumulation_steps=1 \ | |
| --learning_rate=1e-6 \ | |
| --lr_scheduler="constant" \ | |
| --lr_warmup_steps=0 \ | |
| --sample_batch_size=6 \ | |
| --max_train_steps=100000 \ | |
| --save_interval=1500 \ | |
| --save_sample_prompt="image of SystemyTrigger girl" \ | |
| --concepts_list="concepts_list.json" \ | |
| --pad_tokens | |