modify split_tokens to not replicate passthrough keys within a single map call to avoid OOM errors
modify split_tokens to not replicate passthrough keys within a single map call to avoid OOM errors

PiperOrigin-RevId: 446267492

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