Matching output with shape in T5 model

Did you find the bug?
Thanks in advance