examples/unconditional_image_generation/train_unconditional.py · f5ccffecf7c9c7b08c39043d1c9528be2c5cfecc · renzhc / diffusers_dcu

Use `accelerate` save & loading hooks to have better checkpoint structure (#2048) · f5ccffec

Patrick von Platen authored Feb 07, 2023



* better accelerated saving

* up

* finish

* finish

* uP

* up

* up

* fix

* Apply suggestions from code review

* correct ema

* Remove @

* up

* Apply suggestions from code review
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Update docs/source/en/training/dreambooth.mdx
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

---------
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

f5ccffec

train_unconditional.py 24.2 KB

Replace train_unconditional.py