Also:
from transformers import (
T5Config,
T5TokenizerFast,
DataCollatorForSeq2Seq,
EvalPrediction,
HfArgumentParser,
PretrainedConfig,
Trainer,
TrainingArguments,
set_seed,
)
from t5_extra_models import T5ForSequenceClassification
And the corresponding replacements in the code.