search

keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras
Other
1k stars 330 forks source link

Not able to build Stable Diffusion model using `KerasCV` #2100

Closed sayannath closed 11 months ago

sayannath commented 11 months ago

Current Behavior:

While working with the Keras Example Fine-tuning Stable Diffusion.

In the colab notebook, it is using KerasCV v0.4.0 and TensorFlow version is 2.14.0. So it is throwing an error while importing KerasCV i.e

Screenshot 2023-10-11 at 12 38 15 PM

PS: I haven't made any changes in the code. Colab Notebook

After installing the latest KerasCV version i.e 0.6.4, it is throwing an error for SimpleTokenizer, DiffusionModel.

Screenshot 2023-10-11 at 1 02 47 PM
jbischof commented 11 months ago

We cleaned up the submodel exports in #2080. See the colab in the description

sayannath commented 11 months ago

Thanks for the response @jbischof. As it is working, I am closing this issue.