Update tokenization_qwen.py
Browse files- tokenization_qwen.py +7 -1
tokenization_qwen.py
CHANGED
@@ -27,6 +27,12 @@ logger = logging.getLogger(__name__)
|
|
27 |
|
28 |
|
29 |
VOCAB_FILES_NAMES = {"vocab_file": "qwen.tiktoken", "ttf": "SimSun.ttf"}
|
|
|
|
|
|
|
|
|
|
|
|
|
30 |
|
31 |
PAT_STR = r"""(?i:'s|'t|'re|'ve|'m|'ll|'d)|[^\r\n\p{L}\p{N}]?\p{L}+|\p{N}| ?[^\s\p{L}\p{N}]+[\r\n]*|\s*[\r\n]+|\s+(?!\S)|\s+"""
|
32 |
ENDOFTEXT = "<|endoftext|>"
|
@@ -515,7 +521,7 @@ class VisImage:
|
|
515 |
class Visualizer:
|
516 |
def __init__(self, img_rgb, metadata=None, scale=1.0):
|
517 |
self.img = np.asarray(img_rgb).clip(0, 255).astype(np.uint8)
|
518 |
-
self.font_path =
|
519 |
self.output = VisImage(self.img, scale=scale)
|
520 |
self.cpu_device = torch.device("cpu")
|
521 |
|
|
|
27 |
|
28 |
|
29 |
VOCAB_FILES_NAMES = {"vocab_file": "qwen.tiktoken", "ttf": "SimSun.ttf"}
|
30 |
+
FONT_PATH = try_to_load_from_cache("Qwen/Qwen-VL-Chat", "SimSun.ttf")
|
31 |
+
if FONT_PATH is None:
|
32 |
+
if not os.path.exists("SimSun.ttf"):
|
33 |
+
ttf = requests.get("https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-VL/assets/SimSun.ttf")
|
34 |
+
open("SimSun.ttf", "wb").write(ttf.content)
|
35 |
+
FONT_PATH = "SimSun.ttf"
|
36 |
|
37 |
PAT_STR = r"""(?i:'s|'t|'re|'ve|'m|'ll|'d)|[^\r\n\p{L}\p{N}]?\p{L}+|\p{N}| ?[^\s\p{L}\p{N}]+[\r\n]*|\s*[\r\n]+|\s+(?!\S)|\s+"""
|
38 |
ENDOFTEXT = "<|endoftext|>"
|
|
|
521 |
class Visualizer:
|
522 |
def __init__(self, img_rgb, metadata=None, scale=1.0):
|
523 |
self.img = np.asarray(img_rgb).clip(0, 255).astype(np.uint8)
|
524 |
+
self.font_path = FONT_PATH
|
525 |
self.output = VisImage(self.img, scale=scale)
|
526 |
self.cpu_device = torch.device("cpu")
|
527 |
|