Fix bug with new transformers versions
- Newer transformers version add a new kwarg to the embeddings. This now supports that kwarg.
- Newer transformers use .isin, which is not supported on MPS, for a very simple comparison. Fixed this by enabling MPS fallback to CPU.