This repository hosts the TFLite models for the KerasCV Stable Diffusion model. The model can be broken into three parts:
- Text encoder
- Image decoder
- Denoiser
For each model, there is an equivalent TFLite model in this repository.
The TFLite models were generated using this repository. This Colab Notebook also shows how to perform inference.
Note that you need to have tf-nightly installed to use these TFLite models using the Python API.