bert output was returning a dict instead of tuple

[quickSilverShanks]

Your fix works. I was just using this code with transformers v4.41.2 and it returns dictionary by default resulting in variable 'o2' taking the key value 'pooler output' instead of tensor.

Thanks!