Disable SDPA (PyTorch 2.0) in the VAE
Diffusers 0.16.0 made sdpa enabled by default in the vae. It had previously been enabled for the UNet, but of course that didn't affect this repo because it uses its own copy of the UNet.
This fix will not work for diffusers 0.16.0, but it will for 0.16.1 (just released today).
Fixes #173.
Thank you for your interest in contributing to Core ML Stable Diffusion! Please review CONTRIBUTING.md first. If you would like to proceed with making a pull request, please indicate your agreement to the terms outlined in CONTRIBUTING.md by checking the box below. If not, please go ahead and fork this repo and make your updates.
We appreciate your interest in the project!
Do not erase the below when submitting your pull request: #########
- [x] I agree to the terms outlined in CONTRIBUTING.md