Allow flax subfolder
First, I'm sorry about this long list of commits :sweat_smile: - I have my fork set up correctly now so this shouldn't happen again.
Second, this change would be very useful for this PR in `diffusers` so that clip can be loaded from a subfolder: https://github.com/huggingface/diffusers/pull/880#discussion_r1004209900
_The documentation is not available anymore as the PR was closed or merged._