Implement Stable Diffusion #18
Description
It would be great if we could support Stable diffusion in addition to Flux as well. I'm not an expert in image diffusion models, but since Flux seems to start dominating the open-source field in that domain, I believe this might be seen as a low priority. With that said, if anyone wants to work on this, there is a diffusion model tutorial in JAXStack that might be helpful.