Transformers How to use NiallRooney/graph-classification with Transformers:
# Load model directly
from transformers import GraphormerForGraphClassification
model = GraphormerForGraphClassification.from_pretrained("NiallRooney/graph-classification", dtype="auto") 
