File size: 278 Bytes
d1b6379 |
1 2 3 4 5 |
# Load model directly
from transformers import AutoTokenizer, GraphormerForGraphClassification
tokenizer = AutoTokenizer.from_pretrained("PedroLancharesSanchez/graph-regression")
model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression") |