Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: creativeml-openrail-m
|
3 |
+
---
|
4 |
+
|
5 |
+
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:
|
6 |
+
|
7 |
+
* Text encoder
|
8 |
+
* Image decoder
|
9 |
+
* Denoiser
|
10 |
+
|
11 |
+
For each model, there is an equivalent TFLite model in this repository.
|
12 |
+
|
13 |
+
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.
|
14 |
+
|
15 |
+
***Note that you need to have `tf-nightly` installed to use these TFLite models using the Python API.***
|