Could you post a small script to reproduce this issue (or to make the code executable)?
I.e. we would need the model arguments to initialize the model as well as the shapes and types of all input tensors.