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.