Unverified Commit 14d989a9 authored by Clémentine Fourrier's avatar Clémentine Fourrier Committed by GitHub
Browse files

Fixes path for Graphormer checkpoint (#21367)

[FIX] path for Graphormer checkpoint
parent 42b60f8b
......@@ -22,7 +22,7 @@ logger = logging.get_logger(__name__)
GRAPHORMER_PRETRAINED_CONFIG_ARCHIVE_MAP = {
# pcqm4mv1 now deprecated
"graphormer-base": "https://huggingface.co/graphormer-base-pcqm4mv2/resolve/main/config.json",
"graphormer-base": "https://huggingface.co/clefourrier/graphormer-base-pcqm4mv2/resolve/main/config.json",
# See all Graphormer models at https://huggingface.co/models?filter=graphormer
}
......
......@@ -380,7 +380,7 @@ class GraphormerModelTest(ModelTesterMixin, unittest.TestCase):
class GraphormerModelIntegrationTest(unittest.TestCase):
@slow
def test_inference_graph_classification(self):
model = GraphormerForGraphClassification.from_pretrained("graphormer-base-pcqm4mv2")
model = GraphormerForGraphClassification.from_pretrained("clefourrier/graphormer-base-pcqm4mv2")
# Actual real graph data from the MUTAG dataset
# fmt: off
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment