This repository hosts the TFLite version of
diffusion model part of KerasCV Stable Diffusion.
Stable Diffusion consists of
decoder, and some glue codes to handl inputs and outputs of each part. The TFLite version of
diffusion model in this repository is built not only with the
diffusion model itself but also TensorFlow operations that takes
unconditional context from
text encoder and generates
latent output should be passed down to the
decoder which is hosted in this repository.
TFLite conversion was based on the
SavedModel from this repository, and TensorFlow version
>= 2.12-nightly was used.
- Downloads last month