Script for saving pytorch model to safetensors

[justinjohn0306]

@SWivid Can you share the script for saving the finetuned pytorch model as a safetensor model? Thanks :)

[SWivid]

just move out checkpoint['ema_model_state_dict'] to checkpoint

and leverage
from safetensors.torch import save_file, load_file