Fix for issue reported here: https://github.com/google-research/text-to-text-transfer-transformer/issues/281
Fix for issue reported here:
https://github.com/google-research/text-to-text-transfer-transformer/issues/281

PiperOrigin-RevId: 318856283

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