File size: 792 Bytes
1f0a662 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
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 |