SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fixes path for Graphormer checkpoint (#21367)

Browse Source 
[FIX] path for Graphormer checkpoint
This commit is contained in:
Clémentine Fourrier
2023-01-30 21:48:04 +01:00
committed by GitHub
parent 42b60f8b02
commit 14d989a91d
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/graphormer/configuration_graphormer.py
View File

@@ -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
}

2
tests/models/graphormer/test_modeling_graphormer.py
View File

@@ -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