from transformers import AutoModelForCausalLM
from safetensors.torch import save_file
model_name = "microsoft/phi-2"
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto")
save_file(model.state_dict(), "ProTalkModel.safetensors")
| 
