Thank you very much for disclosing the code, is the train_dreambooth.py file missing under the training file please?