how to save my model?

[zjk000]

Hi, how can I save my model so that I can load it in a transformers way? for example:

from transformers import AutoModelForMaskedLM
model = AutoModelForMaskedLM.from_pretrained(the model fine-tuned with openprompt)

[BAOOOOOM]

torch.save?

[fade-color]

here: https://opendelta.readthedocs.io/en/latest/notes/saveload.html