This repository hosts the TFLite version of text encoder 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 text encoder in this repository is built not only with the text encoder itself but also TensorFlow operations that generates context and unconditional context. These output should be passed down to the diffusion model which is hosted in this repository.
TFLite conversion was based on the SavedModel from this repository, and TensorFlow version >= 2.12-nightly was used.
- NOTE: Dynamic range quantization was used.
- NOTE: TensorFlow version
< 2.12-nightlywill fail for the conversion process. - NOTE: For those who wonder how
SavedModelis constructed, find it in keras-sd-serving repository.
- Downloads last month
- 0
Inference API (serverless) does not yet support keras models for this pipeline type.