In the model plan, you can set the model name as GraphTransformer to use the Graph Transformer architecture.
GraphTransformer
model_architecture: model: - name: GraphTransformer