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

..

train_controlnet_webdataset.py

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

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