update
Browse files- tokenization_qwen.py +2 -1
tokenization_qwen.py
CHANGED
@@ -18,6 +18,7 @@ from PIL import Image
|
|
18 |
from PIL import ImageFont
|
19 |
from PIL import ImageDraw
|
20 |
from transformers import PreTrainedTokenizer, AddedToken
|
|
|
21 |
|
22 |
import matplotlib.pyplot as plt
|
23 |
import matplotlib.colors as mcolors
|
@@ -493,7 +494,7 @@ class VisImage:
|
|
493 |
class Visualizer:
|
494 |
def __init__(self, img_rgb, font_path, metadata=None, scale=1.0):
|
495 |
self.img = np.asarray(img_rgb).clip(0, 255).astype(np.uint8)
|
496 |
-
self.font_path =
|
497 |
self.output = VisImage(self.img, scale=scale)
|
498 |
self.cpu_device = torch.device("cpu")
|
499 |
|
|
|
18 |
from PIL import ImageFont
|
19 |
from PIL import ImageDraw
|
20 |
from transformers import PreTrainedTokenizer, AddedToken
|
21 |
+
from transformers.utils import try_to_load_from_cache
|
22 |
|
23 |
import matplotlib.pyplot as plt
|
24 |
import matplotlib.colors as mcolors
|
|
|
494 |
class Visualizer:
|
495 |
def __init__(self, img_rgb, font_path, metadata=None, scale=1.0):
|
496 |
self.img = np.asarray(img_rgb).clip(0, 255).astype(np.uint8)
|
497 |
+
self.font_path = try_to_load_from_cache("Qwen/Qwen-VL-Chat", "SimSun.ttf")
|
498 |
self.output = VisImage(self.img, scale=scale)
|
499 |
self.cpu_device = torch.device("cpu")
|
500 |
|