|
|
--- |
|
|
license: creativeml-openrail-m |
|
|
--- |
|
|
|
|
|
This repository hosts the TFLite models for the [KerasCV Stable Diffusion model](https://github.com/keras-team/keras-cv/blob/master/keras_cv/models/stable_diffusion). 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](https://github.com/sayakpaul/Adventures-in-TensorFlow-Lite/blob/master/Stable_Diffusion_to_TFLite.ipynb). 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.*** |
