Spaces:
Runtime error
Runtime error
Update supertoken.py
Browse files- 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 |
-
|
| 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")
|