pipeline.save_pretrained("dreambooth-pipeline")

Training on a 16GB GPU
	
With the help of gradient checkpointing and the 8-bit optimizer from bitsandbytes, it’s possible to train dreambooth on a 16GB GPU.


	Copied
pip install bitsandbytes
Then pass the --use_8bit_adam option to the training script.


	Copied
export MODEL_NAME="CompVis/stable-diffusion-v1-4"
export INSTANCE_DIR="path_to_training_images"
export CLASS_DIR="path_to_class_images"
