[docs] add the Stable diffusion with Jax/Flax Guide into the docs #2487
Conversation
yiyixuxu
changed the title
|
The documentation is not available anymore as the PR was closed or merged. |
|
Maybe we can also move the JAX/Flax section on the GitHub |
There was a problem hiding this comment.
Cool, @yiyixuxu could you test this once to make sure everything works with current diffusers version (0.14.0) ?
|
And yes we should definitely add |
Co-authored-by: Patrick von Platen <[email protected]>
|
@patrickvonplaten I think flax stopped working on colab again 😭😭😭 |
|
@stevhliu It seems that Flax/Jax section in README only contains the most basic use examples - I think these examples needed to be added to docstrings so they would show up on the flax pipelines' API page. What do you think? |
|
Big yes to both of your questions @yiyixuxu 👍. Having all the |
hey i've been wrestling with this specific issue (Flax HF pipelines on Colab) for weeks and finally got past the version compatibility issue, at least with Flax; check the workaround in the issue post here - might be useful but i'm not sure how the automated setup of Diffusers/Transformers dependencies would handle the recursive dependency of Flax<>Orbax; i.e. if my workaround method could be used here, or required to be done before installing HF libs... |
|
Gentle ping here @yiyixuxu |
…ggingface#2487) * add stable diffusion jax guide --------- Co-authored-by: Patrick von Platen <[email protected]>
…ggingface#2487) * add stable diffusion jax guide --------- Co-authored-by: Patrick von Platen <[email protected]>
No description provided.