'T5ForConditionalGeneration' object has no attribute '_prune_heads'

Hi @ceddyla,

Did you figure out a solution for this problem?

I’m facing same issue and raised a Github issue here - https://github.com/huggingface/transformers/issues/27044