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")