diffusers

mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Files

Bagheera 8e963d1c2a 7529 do not disable autocast for cuda devices (#7530 )

* 7529 do not disable autocast for cuda devices

* Remove typecasting error check for non-mps platforms, as a correct autocast implementation makes it a non-issue

* add autocast fix to other training examples

* disable native_amp for dreambooth (sdxl)

* disable native_amp for pix2pix (sdxl)

* remove tests from remaining files

* disable native_amp on huggingface accelerator for every training example that uses it

* convert more usages of autocast to nullcontext, make style fixes

* make style fixes

* style.

* Empty-Commit

---------

Co-authored-by: bghira <bghira@users.github.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

2024-04-02 20:15:06 +05:30

text_to_image

7529 do not disable autocast for cuda devices (#7530 )

2024-04-02 20:15:06 +05:30

textual_inversion

7529 do not disable autocast for cuda devices (#7530 )

2024-04-02 20:15:06 +05:30

unconditional_image_generation

7529 do not disable autocast for cuda devices (#7530 )

2024-04-02 20:15:06 +05:30

README.md

[Docs] Fix typos (#6122 )

2023-12-11 10:55:28 -08:00

README.md

Diffusers examples with ONNXRuntime optimizations

This research project is not actively maintained by the diffusers team. For any questions or comments, please contact Prathik Rao (prathikr), Sunghoon Choi (hanbitmyths), Ashwini Khade (askhade), or Peng Wang (pengwa) on github with any questions.

This aims to provide diffusers examples with ONNXRuntime optimizations for training/fine-tuning unconditional image generation, text to image, and textual inversion. Please see individual directories for more details on how to run each task using ONNXRuntime.