sayakpaul HF staff commited on
Commit
a877e01
·
1 Parent(s): bedf134

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
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.***