fix safetensor saving (#1954)

Commit

2 years ago

fix safetensor saving (#1954) * fix safetensor saving * fix test * fix * better save * modify as keyword arg