Adding `safetensors` variant of this model (#1)

add modifying codes

revise the checkpoint in official repo; and make it fit in huggingface arch

initial commit