RuntimeError: output with shape [64, 12, 1, 1] doesn't match the broadcast shape [64, 12, 1, 64]

I’m trying to make a project on English to SQL Query, I’m using T5-base model and while evaluating the model I ran into this error.

Here are my basic configurations:

define constants

MODEL_NAME = ‘t5-base’
MAX_LENGTH = 64
BATCH_SIZE = 64
NUM_EPOCHS = 1

I’m attaching the notebook link:

Please review the code and let me know where I’m doing wrong.
TIA