In this video, I'll show you how to train amazing dreambooth models with the newly released SDXL 1.0! In addition to that, we will also learn how to generate images using SDXL base model and the use of refiner to enhance the quality of generated images.
The commands used for training in this video are as follows:
!pip install -U autotrain-advanced
!autotrain setup --update-torch
!autotrain dreambooth \
--model stabilityai/stable-diffusion-xl-base-1.0 \
--output output/ \
--image-path images/ \
--prompt "photo of sks dog" \
--resolution 1024 \
--batch-size 1 \
--num-steps 500 \
--fp16 \
--gradient-accumulation 4 \
--lr 1e-4
On google colab, you can add --use-8bit-adam parameter and change resolution to 512 if you are on free version of google colab.
Inference code:
Please subscribe and like the video to help me keep motivated to make awesome videos like this one. :)
My book, Approaching (Almost) Any Machine Learning problem, is available for free here: bit.ly/approachingml
Follow me on:
Twitter: / abhi1thakur
LinkedIn: / abhi1thakur
Kaggle: kaggle.com/abhishek
Негізгі бет Stable Diffusion XL (SDXL) DreamBooth: Easy, Fast & Free | Beginner Friendly
Пікірлер: 136