diffusers
diffusers copied to clipboard
Add v-prediction to Flax training examples
Added v-prediction to FlaxDDIM and FlaxDDPM schedulers as well as the Flax training examples.
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint.
Gently ping @pcuenca and @yiyixuxu here
We'll also need to resolve the conflict and rebase to main. Let me know if you need any help.