`datasets.map` calls a function that requires a `transformers.PreTrainedModel` object - unpickable object

🤗Datasets
2

Oh and, if that’s useful, here’s my env:

- `datasets` version: 2.7.0
- Platform: Linux-5.4.0-1093-gcp-x86_64-with-glibc2.17
- Python version: 3.8.13
- PyArrow version: 7.0.0
- Pandas version: 1.5.0