Adds Flax checkpoint
#4
by
afmck - opened
No description provided.
afmck changed pull request status to
open
Thanks @afmck ! cc @young-geng and @ haoliu - would one of you kindly be able to merge the Flax weights for open_llama_3b_v2? This is part of the ongoing integration: https://github.com/huggingface/transformers/pull/24587