asa1108 commited on
Commit
88d1580
·
verified ·
1 Parent(s): f77d13c

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. train.sh +1 -1
train.sh CHANGED
@@ -1 +1 @@
1
- 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/rhinoplasty-lora --output_dir=/home/user/app/experiments/rhinoplasty-lora --instance_prompt="my_rhinoplasty" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=1000 --checkpointing_steps=100 --validation_prompt="my_rhinoplasty" --validation_epochs=100 --seed=0
 
1
+ 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/rhinoplasty-lora --output_dir=/home/user/app/experiments/rhinoplasty-lora --instance_prompt="my_rhinoplasty" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=1000 --checkpointing_steps=100 --validation_prompt="my_rhinoplasty" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam