flax
@nitrosocke I converted to flax
it is here https://huggingface.co/camenduru/mo-di/tree/flax I think if we push here it will be better
this PR not going to flax branch π I asked to @patrickvonplaten I will wait his answer
Alright, thank you for the update. I'm sure we can get this set up.
Hey @camenduru ,
I'm looking into it - will try to answer shortly!
Ok for now there is sadly no way really for community contributors (cc @julien-c )
However, we should aim for flax and pytorch weigths to live both on the main branch which will be enabled here: https://github.com/huggingface/diffusers/pull/1107
@camenduru
could you maybe open a PR to add flax weigts to the "main" repo?
@nitrosocke
- this shouldn't make the download of the weights any slower in diffusers
and is probably the easiest way to deal with it.
@nitrosocke
if you really prefer to add it to a "flax" branch however, I'm happy to help with it.
Thank you
@patrickvonplaten
for looking into it.
I would prefer to add the flux branch to keep things organised, as some users are already confused with the ckpt file and the diffusers being in the same folder.
thanks @patrickvonplaten , all in one place good idea. π€© I converted more and put them here https://huggingface.co/flax I don't have permission to add in flax branch please copy paste π