Modelpublic

prelington/CodexTrouter

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face

1likes

ProTalk_ModelExporter.py7 linesDownload Raw Back to root

1from transformers import AutoModelForCausalLM2from safetensors.torch import save_file3 4model_name = "microsoft/phi-2"5model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto")6save_file(model.state_dict(), "ProTalkModel.safetensors")7