annoyingball / train.sh
Zephyrus-H's picture
Upload . with huggingface_hub
4cef2fa verified
Raw
History Blame Contribute Delete
566 Bytes
accelerate launch train_dreambooth_lora.py --pretrained_model_name_or_path=stabilityai/stable-diffusion-2-1-base --instance_data_dir=/home/user/app/training_data/annoyingball --output_dir=/home/user/app/experiments/annoyingball --instance_prompt="<Annoyingball>" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=4 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=1000 --checkpointing_steps=100 --validation_prompt="an <Annoyingball> is sitting on a chair" --validation_epochs=100 --seed=0 --mixed_precision fp16