This repository implements the Dreambooth fine-tuning approach from Google’s research on Stable Diffusion, enabling users to personalize image generation models with just a few example images. The implementation extends Textual Inversion by fine-tuning the entire diffusion model rather than just word embeddings, using PyTorch and PyTorch Lightning for training on GPU.