Adding `safetensors` variant of this model

upload flax model

allow flax

first default model

initial commit