Hello. I’m trying to export .pt2 from the model of transformers library.
But, I’ve got module object not callable error.
How can I call the export api for this case?
import torch
from transformers import T5Tokenizer, T5Model
model = T5Model.from_pretrained("t5-small")
ep = torch.export(model, model.dummy_inputs)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: 'module' object is not callable