Add vocabulary argument to maintain compatibility with other dataset fns.
Add vocabulary argument to maintain compatibility with other dataset fns.

PiperOrigin-RevId: 338567988

google-research/text-to-text-transfer-transformer