examples/text_to_image/train_text_to_image.py · f5ccffecf7c9c7b08c39043d1c9528be2c5cfecc · OpenDAS / diffusers

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_text_to_image.py 30.9 KB

Replace train_text_to_image.py