nicoladecao
commited on
Merge branch 'main' of https://huggingface.co/facebook/genre-linking-aidayago2 into main
Browse files
README.md
CHANGED
@@ -46,8 +46,8 @@ import pickle
|
|
46 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
47 |
|
48 |
# OPTIONAL: load the prefix tree (trie), you need to additionally download
|
49 |
-
# https://huggingface.co/facebook/genre-
|
50 |
-
# https://huggingface.co/facebook/genre-
|
51 |
# from trie import Trie
|
52 |
# with open("kilt_titles_trie_dict.pkl", "rb") as f:
|
53 |
# trie = Trie.load_from_dict(pickle.load(f))
|
|
|
46 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
47 |
|
48 |
# OPTIONAL: load the prefix tree (trie), you need to additionally download
|
49 |
+
# https://huggingface.co/facebook/genre-linking-aidayago2/blob/main/trie.py and
|
50 |
+
# https://huggingface.co/facebook/genre-linking-aidayago2/blob/main/kilt_titles_trie_dict.pkl
|
51 |
# from trie import Trie
|
52 |
# with open("kilt_titles_trie_dict.pkl", "rb") as f:
|
53 |
# trie = Trie.load_from_dict(pickle.load(f))
|