Remove fallback option of configuring loss method directly rather than on the model class constructor.
Remove fallback option of configuring loss method directly rather than on the model class constructor.

PiperOrigin-RevId: 432423468

google-research/t5x