Commit
•
32386e3
1
Parent(s):
7d15402
Update README.md
Browse files
README.md
CHANGED
|
@@ -43,10 +43,11 @@ Here is an example of generation for Wikipedia page retrieval for open-domain fa
|
|
| 43 |
|
| 44 |
```python
|
| 45 |
import pickle
|
| 46 |
-
from trie import Trie
|
| 47 |
from transformers import BartTokenizer, BartForConditionalGeneration
|
| 48 |
|
| 49 |
-
# OPTIONAL: load the prefix tree (trie)
|
|
|
|
|
|
|
| 50 |
# with open("kilt_titles_trie_dict.pkl", "rb") as f:
|
| 51 |
# trie = Trie.load_from_dict(pickle.load(f))
|
| 52 |
|
|
|
|
| 43 |
|
| 44 |
```python
|
| 45 |
import pickle
|
|
|
|
| 46 |
from transformers import BartTokenizer, BartForConditionalGeneration
|
| 47 |
|
| 48 |
+
# OPTIONAL: load the prefix tree (trie), you need to download trie.py from
|
| 49 |
+
# https://huggingface.co/facebook/genre-kilt/blob/main/trie.py
|
| 50 |
+
# from trie import Trie
|
| 51 |
# with open("kilt_titles_trie_dict.pkl", "rb") as f:
|
| 52 |
# trie = Trie.load_from_dict(pickle.load(f))
|
| 53 |
|