TypeError: __init__() got an unexpected keyword argument 'gradient_checkpointing'

Question

this is the error and
the code

I have been trying to train the model using one of my own model as a pretrained model and every time I run, I get this error. The moment I use the previously used pretrained model other than my own one, I don't get that error. Please help me fix this error

Answer 1

your config.json file might have missed "gradient_checkpointing": true, I think this will help.