diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard
Published 1 month ago • verified publisher icon huggingface

TensorFlow and Flax/Jax

Open will-rice opened this issue 3 years ago • 1 comments

Will this library be similar to transformers and include TensorFlow and Flax implementations at some point?

will-rice avatar Jul 28 '22 14:07 will-rice

Yes that is on the longer-term road map. Moderate term we want to make the schedulers interface well with numpy (which then lets one use their own non-torch models), but making the models all independent is more substantial of an undertaking.

We would happily provide feedback if you want to open a PR!

natolambert avatar Jul 28 '22 14:07 natolambert

We're currently adding JAX with see https://github.com/huggingface/diffusers/issues/475 . We don't plan on adding TF in anytime soon.

patrickvonplaten avatar Sep 13 '22 15:09 patrickvonplaten