ValueError: Data cardinality is ambiguous: x sizes: 1, 500, 500 Please provide data which shares the same first dimension

Question

My seq2seq model fits on data perfectly, but the predict method on inference model gives the above error when i pass the start token do the model initialized with the encoder st