Advertisement
ZeroCool22

DB Train

Oct 14th, 2022
238
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.84 KB | None | 0 0
  1. ***My path folder (WSL2 + Ubuntu):***
  2.  
  3. https://i.imgur.com/UwzYmxK.jpg
  4.  
  5. ***My train config file:***
  6.  
  7. export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH
  8. export MODEL_NAME="CompVis/stable-diffusion-v1-4"
  9. export INSTANCE_DIR="training"
  10. export CLASS_DIR="classes"
  11. export OUTPUT_DIR="my_model"
  12.  
  13. accelerate launch train_dreambooth.py \
  14. --pretrained_model_name_or_path=$MODEL_NAME \
  15. --instance_data_dir=$INSTANCE_DIR \
  16. --class_data_dir=$CLASS_DIR \
  17. --output_dir=$OUTPUT_DIR \
  18. --with_prior_preservation --prior_loss_weight=1.0 \
  19. --instance_prompt="AlbertoFinstance" \
  20. --class_prompt="man" \
  21. --resolution=512 \
  22. --train_batch_size=1 \
  23. --learning_rate=5e-6 \
  24. --lr_scheduler="constant" \
  25. --lr_warmup_steps=0 \
  26. --gradient_accumulation_steps=1 --gradient_checkpointing \
  27. --use_8bit_adam \
  28. --max_train_steps=2500
  29.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement