Adding `safetensors` variant of this model

upload flax model

allow flax

Update pytorch_model.bin

Update config.json

initial commit