Hello, due to the changes of lightening.Trainer.add_argparse_args method: https://github.com/Lightning-AI/lightning/pull/16708 I guess implementations using add_argparse_args like this: https://github.com/XavierXiao/Dreambooth-Stable-Diffusion/blob/main/main.py#L554 should be changed. If there is any misunderstanding, please let me know. Thank you.