I am trying to build an encoder and decoder using LSTM with attention layer. I am getting an Assertion Error in the /content/attention_keras/layers/attention.py layer as shown b