T5 version

[dawei03896]

thanks for sharing this good project, does it support other version T5 models, like ByT5 Base, ByT5 Small?