File size: 692 Bytes
a877e01 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
---
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.*** |