• Bagheera's avatar
    7529 do not disable autocast for cuda devices (#7530) · 8e963d1c
    Bagheera authored
    
    
    * 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: default avatarbghira <bghira@users.github.com>
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    8e963d1c
train_custom_diffusion.py 58.4 KB