| 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") | |