[Python] Saving transformers model with save_pretrained - tuple error

I am running the following commands in Jupyter notebook:

!pip install -U transformers


.

from transformers import AutoModel, AutoTokenizer
model = AutoModel.from_pretrained("microsoft/deberta-v3-large", torch_dtype="auto")
tokenizer = AutoTokenizer.from_pretrained("microsoft/deberta-v3-large")


Then I am trying to save it locally:

save_directory = "/some_local_folder/"
model.save_pretrained(save_directory)


It generates the following error:

---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
/tmp/ipykernel_36/453108176.py in <cell line: 0>()
1 save_directory = "/some_local_folder/" # Replace with your desired path
----> 2 model.save_pretrained(save_directory)

AttributeError: 'tuple' object has no attribute 'save_pretrained'


Error makes sense, since when check type(model) it is indeed tuple. So am I loading it incorrectly? How can save it locally?

Continue reading...