gsaltintas commited on
Commit
cf73d0b
·
verified ·
1 Parent(s): b1c588a

Update supertoken.py

Browse files
Files changed (1) hide show
  1. supertoken.py +1 -8
supertoken.py CHANGED
@@ -17,8 +17,6 @@ from typing import List, Optional
17
  import tokenizers
18
  import transformers
19
  import yaml
20
- from xarch_tokenizers.models import load_tokenizer as hf_load_tokenizer
21
- from xarch_tokenizers.utils import system
22
 
23
  Vocab = dict[str, list[int]]
24
 
@@ -112,12 +110,7 @@ class HFTokenizer(Tokenizer):
112
 
113
  @classmethod
114
  def load(cls, name):
115
- if system.get_host() == system.Hosts.vector:
116
- name = system.VECTOR_HF_MAPPING.get(name, name)
117
- try:
118
- tok = hf_load_tokenizer(name)
119
- except:
120
- tok = transformers.AutoTokenizer.from_pretrained(name)
121
  sts = getattr(tok, "special_tokens_map", {})
122
  if "bert" in name:
123
  bos_str = sts.get("cls_token")
 
17
  import tokenizers
18
  import transformers
19
  import yaml
 
 
20
 
21
  Vocab = dict[str, list[int]]
22
 
 
110
 
111
  @classmethod
112
  def load(cls, name):
113
+ tok = transformers.AutoTokenizer.from_pretrained(name)
 
 
 
 
 
114
  sts = getattr(tok, "special_tokens_map", {})
115
  if "bert" in name:
116
  bos_str = sts.get("cls_token")