From 14d989a91d8a43154b2e3c7abadd5d3cb721e5b2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9mentine=20Fourrier?=
 <22726840+clefourrier@users.noreply.github.com>
Date: Mon, 30 Jan 2023 21:48:04 +0100
Subject: [PATCH] Fixes path for Graphormer checkpoint (#21367)

[FIX] path for Graphormer checkpoint
---
 src/transformers/models/graphormer/configuration_graphormer.py | 2 +-
 tests/models/graphormer/test_modeling_graphormer.py            | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/transformers/models/graphormer/configuration_graphormer.py b/src/transformers/models/graphormer/configuration_graphormer.py
index 72e2adb836..17791bfb0c 100644
--- a/src/transformers/models/graphormer/configuration_graphormer.py
+++ b/src/transformers/models/graphormer/configuration_graphormer.py
@@ -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
 }
 
diff --git a/tests/models/graphormer/test_modeling_graphormer.py b/tests/models/graphormer/test_modeling_graphormer.py
index 52f5604a50..ed692c8868 100644
--- a/tests/models/graphormer/test_modeling_graphormer.py
+++ b/tests/models/graphormer/test_modeling_graphormer.py
@@ -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