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