keras-sd/

diffusion-model-tflite

This repository hosts the TFLite version of diffusion model part of KerasCV Stable Diffusion.

Stable Diffusion consists of text encoder, diffusion model, 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 context, unconditional context from text encoder and generates latent. The 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.