Adding `safetensors` variant of this model

allow flax

commit from juliame

initial commit