Advertisement
Arrowman6677
Nov 3rd, 2022
56
0
Never
This is comment for paste StableDiffusion-WSL
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. I had to make some changes to get this to work for me:
  2.  
  3. * Install required packages (updated for 11/02/2022)
  4. pip install .
  5. cd examples/dreambooth
  6. conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge
  7. pip install -r requirements.txt
  8. pip install -U --pre triton
  9. pip install ninja bitsandbytes
  10. conda install xformers -c xformers/label/dev
  11.  
  12. I changed the torch install to use 0.12 (the OG pastebin installs the latest 0.13, which did not work for me). I also changed xformers to grab the latest version.
  13.  
  14. * Training file (running with SD 1.5)
  15. 1. Get your training script from here (https://github.com/ShivamShrirao/diffusers/tree/main/examples/dreambooth). I used the 8gb one and followed the instructions for configuring DeepSpeed
  16. 2. make sure you save your training script with LF line separators and not CRLF
  17. 3. Follow the instructions from this step to dump your 1.5 model (https://github.com/ShivamShrirao/diffusers/issues/50#issuecomment-1294854643).
  18. 4. Update the training script for the new model. It should look something like:
  19.  
  20. export MODEL_DIR="/home/arrow/github/diffusers/models/dump"
  21. export VAE_DIR="/home/arrow/github/diffusers/models/dump/vae"
  22. export INSTANCE_DIR="training"
  23. export CLASS_DIR="classes"
  24. export OUTPUT_DIR="out"
  25.  
  26. accelerate launch train_dreambooth.py \
  27. --pretrained_model_name_or_path=$MODEL_DIR \
  28. --pretrained_vae_name_or_path=$VAE_DIR \
  29. ...
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement