Fixes path for Graphormer checkpoint (#21367)

[FIX] path for Graphormer checkpoint
2023-01-30 21:48:04 +01:00
parent 42b60f8b02
commit 14d989a91d
2 changed files with 2 additions and 2 deletions
--- 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