Add Flax weights

#5
by ArthurZ HF Staff - opened

Fix weights after#17785

ArthurZ changed pull request status to merged

Sign up or log in to comment