Error on DistilBertForQuestionAnswering.from_pretrained()
#5
by
ravneetg
- opened
I have this piece of code that is giving me a error:
model_name = "distilbert-base-uncased-distilled-squad"
model = DistilBertForQuestionAnswering.from_pretrained(model_name)
Error I am getting is: TypeError: register_buffer() got an unexpected keyword argument 'persistent'
Any pointers on what am I missing?