Pedro Cuenca 449a16d96e Disable SDPA (PyTorch 2.0) in the VAE (#174) ...

* 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 it 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 main or
0.16.1 (to be released today).

* Link to coremltools issue.

2023-04-28 21:14:25 -07:00

..

__init__.py

Initial commit

2022-11-30 13:41:05 -06:00

_version.py

Initial commit

2022-11-30 13:41:05 -06:00

chunk_mlprogram.py

Initial commit

2022-11-30 13:41:05 -06:00

controlnet.py

Support ControlNet (#153)

2023-04-18 16:17:59 -07:00

coreml_model.py

Support ControlNet (#153)

2023-04-18 16:17:59 -07:00

layer_norm.py

Initial commit

2022-11-30 13:41:05 -06:00

pipeline.py

Fixes #160

2023-04-24 12:13:30 -07:00

torch2coreml.py

Disable SDPA (PyTorch 2.0) in the VAE (#174)

2023-04-28 21:14:25 -07:00

unet.py

Support ControlNet (#153)

2023-04-18 16:17:59 -07:00