Huggingface模型转PyTorch模型

from transformers import AutoTokenizer
from transformers import AutoModel
import torch
from transformers import LlamaTokenizer, LlamaForCausalLM

# convert huggingface model to pytorch model

modelfromhuggingface=LlamaForCausalLM.from_pretrained(model_name_from_huggingface)
print("model from huggingface loaded, now eval")
modelfromhuggingface.eval()
print("eval finished,now convert")
torch.save(modelfromhuggingface, 'model_path_and_name.pt')
print("convert finished")