Add Flax weights

#7
by ArthurZ HF staff - opened

Fix weights after#17785

ArthurZ changed pull request status to merged

Sign up or log in to comment