'T5ForConditionalGeneration' object has no attribute '_prune_heads'

Hi guys

I have been following the documentation here to prune heads but keep getting this error AttributeError: ‘T5ForConditionalGeneration’ object has no attribute ‘_prune_heads’

This is my code, I was just following the docs


c