How do you convert weights to flax?

by jffacevedo - opened Mar 8

Mar 8

@ariG23498 can you tell me how you convert the weights from pytorch to flax? Thanks!

Owner Mar 10

@jffacevedo I had probably used FLAX-T5 and the Torch checkpoint to convert the model on the fly.

If that does not work, let me know.

Mar 10

ok, let me try that. Thanks!

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment