Adding `safetensors` variant of this model

allow flax

first version

initial commit