Hey !
I have the minimal example in this repo:https://github.com/sudarshan1994/create_graph_debug 19
Its easy to run : python test.py and does not call any weird libraries haha !
The pre-trained model is also uploaded in the repo, and its essential for seeing the discrepancy. For randomly initialized models we dont see the discrepancy.
The whole thing takes a minute to run ! All the relevant action takes place between line 146 to line 170, the rest is just setting up the Bert model.
Thanks a lot for agreeing to take a look at the code.