That’s a bit unexpected, as the error claims a 1D activation is used at one point:
but got 1-dimensional input of size [128] instead

Could you post the shape of the input to the model as well as the setup you are using (model initialization etc.) so that we could reproduce and debug it?