keras-team / keras-hub

Pretrained model hub for Keras 3
Apache License 2.0
803 stars 243 forks source link

Add StableDiffusion3 preset #1884

Closed james77777778 closed 2 months ago

james77777778 commented 2 months ago

This PR adapts #1860 with minor updates to support config_name in preprocessor as well. This is neccssary for SD3, as it has a nested preprocessor.

A small bug in CLIPTokenizer and CLIPPreprocessor has been addressed.

A demo colab showcasing a very clean API for generating new images: https://colab.research.google.com/drive/1ucx2lEck1ZO3HVTjZ4qgVsQrBWrHWfVQ?usp=sharing

https://www.kaggle.com/models/kerashub/stablediffusion3

@divyashreepathihalli @mattdangerw @SamanehSaadat