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)
|