search

XavierXiao / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion
MIT License
7.6k stars 795 forks source link

Interface changed for add_argparse_args() of lightning.Trainer #143

Open skspark opened 1 year ago

skspark commented 1 year ago

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.

MingfeiShiMS commented 1 year ago

This also happens to me for pytorch-lightning==2.0.7, and when I degrade the version to 1.9.0, it works. I hope this helps you.