Without inference model, I am able to train the model. But when I write code for inference with attention following error in the attached image pops. Kindly help me with his
