Skip to content

Commit

Permalink
Update examples.ipynb
Browse files Browse the repository at this point in the history
  • Loading branch information
nicola-decao authored Jun 10, 2022
1 parent e8c047d commit a7d645c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples_mgenre/examples.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -22,10 +22,10 @@
"* optionally, we can use a **mention table** to restrict the search space to a number of candidates [mention2wikidataID_with_titles_label_alias_redirect](https://dl.fbaipublicfiles.com/GENRE/mention2wikidataID_with_titles_label_alias_redirect.pkl).\n",
"\n",
"\n",
"# mGENRE for transformers (coming soon!)\n",
"# mGENRE for transformers\n",
"\n",
"First make sure that you have [transformers](https://github.com/huggingface/transformers) >=4.2.0 installed. \n",
"**NOTE: we used fairseq for all experiments in the paper. The huggingface/transformers models are obtained with a conversion script similar to [this](https://github.com/huggingface/transformers/blob/master/src/transformers/models/bart/convert_bart_original_pytorch_checkpoint_to_pytorch.py).**\n",
"**NOTE: we used fairseq for all experiments in the paper. The huggingface/transformers models are obtained with a [conversion script](https://github.com/facebookresearch/GENRE/blob/main/scripts_genre/convert_bart_original_pytorch_checkpoint_to_pytorch.py).**\n",
"\n",
"Then load the trie and define the function to apply the constraints with the entities trie"
]
Expand Down

0 comments on commit a7d645c

Please sign in to comment.