fix ttf path

tokenization_qwen.py

@@ -169,6 +169,9 @@ class QWenTokenizer(PreTrainedTokenizer):
         self.im_start_id = self.special_tokens[IMSTART]
         self.im_end_id = self.special_tokens[IMEND]
 
+        model_dir = getattr(self, 'model_dir', '')
+        self.font_path = os.path.join(model_dir, "SimSun.ttf")
+
     def __len__(self) -> int:
         return self.tokenizer.n_vocab
 
@@ -415,13 +418,11 @@ class QWenTokenizer(PreTrainedTokenizer):
         else:
             image = plt.imread(image)
             h, w = image.shape[0], image.shape[1]
-        visualizer = Visualizer(image)
+        visualizer = Visualizer(image, font_path=self.font_path)
 
         boxes = self._fetch_all_box_with_ref(response)
         if not boxes:
             return None
-        # fnt = ImageFont.truetype("SimSun.ttf", 50)
-        # draw = ImageDraw.Draw(image)
         color = random.choice([_ for _ in mcolors.TABLEAU_COLORS.keys()]) # init color
         for box in boxes:
             if 'ref' in box: # random new color for new refexps
@@ -492,8 +493,9 @@ class VisImage:
 
 
 class Visualizer:
-    def __init__(self, img_rgb, metadata=None, scale=1.0):
+    def __init__(self, img_rgb, font_path, metadata=None, scale=1.0):
         self.img = np.asarray(img_rgb).clip(0, 255).astype(np.uint8)
+        self.font_path = font_path
         self.output = VisImage(self.img, scale=scale)
         self.cpu_device = torch.device("cpu")
 
@@ -525,7 +527,7 @@ class Visualizer:
             y,
             text,
             size=font_size * self.output.scale,
-            fontproperties=FontProperties(fname=r"SimSun.ttf"),
+            fontproperties=FontProperties(fname=self.font_path),
             bbox={"facecolor": "black", "alpha": 0.8, "pad": 0.7, "edgecolor": "none"},
             verticalalignment="top",
             horizontalalignment=horizontal_alignment,