I would recommend to use the same workflow as before.
Once you see the shape mismatch, print the model, search for the layer which is creating the error, and increase its in_features to the needed value, which is given in the error message.