Adding `safetensors` variant of this model

upload flax model

allow flax

add updated model files

remove model files

add model files

initial commit