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__) ...@@ -22,7 +22,7 @@ logger = logging.get_logger(__name__)
GRAPHORMER_PRETRAINED_CONFIG_ARCHIVE_MAP = { GRAPHORMER_PRETRAINED_CONFIG_ARCHIVE_MAP = {
# pcqm4mv1 now deprecated # 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 # See all Graphormer models at https://huggingface.co/models?filter=graphormer
} }
......
...@@ -380,7 +380,7 @@ class GraphormerModelTest(ModelTesterMixin, unittest.TestCase): ...@@ -380,7 +380,7 @@ class GraphormerModelTest(ModelTesterMixin, unittest.TestCase):
class GraphormerModelIntegrationTest(unittest.TestCase): class GraphormerModelIntegrationTest(unittest.TestCase):
@slow @slow
def test_inference_graph_classification(self): 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 # Actual real graph data from the MUTAG dataset
# fmt: off # 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