Add option to choose T5 model size.
I believe the error mentioned in #3469 is due to the example tests loading T5-large in memory. One of the workers load that model which fills up the machine's memory, and other workers crash with an oom error.

This PR gives the option to choose the T5 model size, and changes the tests to only use the small model.