nicoladecao commited on
Commit
2695201
1 Parent(s): bd735c2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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-kilt/blob/main/trie.py and
50
- # https://huggingface.co/facebook/genre-kilt/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))
 
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))