From https://github.com/jinzishuai/ray/blob/master/doc/source/templates/05_dreambooth_finetuning