Adding `safetensors` variant of this model

upload flax model

allow flax

checking in the initial version files

initial commit