Adding `safetensors` variant of this model

updates all files

small fix

fix figs

add figures

updates readme

add running codes

First model version

initial commit