fix
Browse files- tokenization_qwen.py +1 -1
tokenization_qwen.py
CHANGED
|
@@ -414,7 +414,7 @@ class QWenTokenizer(PreTrainedTokenizer):
|
|
| 414 |
h, w = image.height, image.width
|
| 415 |
else:
|
| 416 |
image = plt.imread(image)
|
| 417 |
-
|
| 418 |
visualizer = Visualizer(image)
|
| 419 |
|
| 420 |
boxes = self._fetch_all_box_with_ref(response)
|
|
|
|
| 414 |
h, w = image.height, image.width
|
| 415 |
else:
|
| 416 |
image = plt.imread(image)
|
| 417 |
+
h, w = image.shape[0], image.shape[1]
|
| 418 |
visualizer = Visualizer(image)
|
| 419 |
|
| 420 |
boxes = self._fetch_all_box_with_ref(response)
|