fix font path
Browse files- tokenization_qwen.py +2 -3
tokenization_qwen.py
CHANGED
|
@@ -169,7 +169,6 @@ class QWenTokenizer(PreTrainedTokenizer):
|
|
| 169 |
self.eod_id = self.tokenizer.eot_token
|
| 170 |
self.im_start_id = self.special_tokens[IMSTART]
|
| 171 |
self.im_end_id = self.special_tokens[IMEND]
|
| 172 |
-
self.font_path = os.path.dirname(__file__)
|
| 173 |
|
| 174 |
def __len__(self) -> int:
|
| 175 |
return self.tokenizer.n_vocab
|
|
@@ -417,7 +416,7 @@ class QWenTokenizer(PreTrainedTokenizer):
|
|
| 417 |
else:
|
| 418 |
image = plt.imread(image)
|
| 419 |
h, w = image.shape[0], image.shape[1]
|
| 420 |
-
visualizer = Visualizer(image
|
| 421 |
|
| 422 |
boxes = self._fetch_all_box_with_ref(response)
|
| 423 |
if not boxes:
|
|
@@ -492,7 +491,7 @@ class VisImage:
|
|
| 492 |
|
| 493 |
|
| 494 |
class Visualizer:
|
| 495 |
-
def __init__(self, img_rgb,
|
| 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)
|
|
|
|
| 169 |
self.eod_id = self.tokenizer.eot_token
|
| 170 |
self.im_start_id = self.special_tokens[IMSTART]
|
| 171 |
self.im_end_id = self.special_tokens[IMEND]
|
|
|
|
| 172 |
|
| 173 |
def __len__(self) -> int:
|
| 174 |
return self.tokenizer.n_vocab
|
|
|
|
| 416 |
else:
|
| 417 |
image = plt.imread(image)
|
| 418 |
h, w = image.shape[0], image.shape[1]
|
| 419 |
+
visualizer = Visualizer(image)
|
| 420 |
|
| 421 |
boxes = self._fetch_all_box_with_ref(response)
|
| 422 |
if not boxes:
|
|
|
|
| 491 |
|
| 492 |
|
| 493 |
class Visualizer:
|
| 494 |
+
def __init__(self, img_rgb, metadata=None, scale=1.0):
|
| 495 |
self.img = np.asarray(img_rgb).clip(0, 255).astype(np.uint8)
|
| 496 |
self.font_path = try_to_load_from_cache("Qwen/Qwen-VL-Chat", "SimSun.ttf")
|
| 497 |
self.output = VisImage(self.img, scale=scale)
|