I have created a model using the function and am passing two embeddings as an input to the model in the form of 1D arrays. While trying to fit Grid search on the inputs, it thro