test2 / train.sh
ClaireOzzz's picture
Upload folder using huggingface_hub
0373ac6
accelerate launch train_dreambooth_lora.py --pretrained_model_name_or_path=stabilityai/stable-diffusion-2-1-base --instance_data_dir=/content/SpatialTerm7/LoRA-DreamBooth-Training-UI/training_data/test2 --output_dir=/content/SpatialTerm7/LoRA-DreamBooth-Training-UI/experiments/test2 --instance_prompt="prrr" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=100 --checkpointing_steps=100 --validation_prompt="vases" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam