Fix Graphormer test suite
# What does this PR do?
Fixes comment in #21367, @ydshieh 
Updated shape of model instantiation when calling pretrained in test suite, + updated values in the integration test.


_The documentation is not available anymore as the PR was closed or merged._