[chameleon] fix num image token check (#36918)

* [chameleon] fix num image token check

* embed after merging image token

* skip this also

* mistral require_read_token
This commit is contained in:
Raushan Turganbay
2025-03-24 12:36:08 +01:00
committed by GitHub
parent a41e08aa19
commit 57f551c78d
4 changed files with 160 additions and 13 deletions

View File

@@ -20,7 +20,7 @@ import unittest
import requests
from transformers import PixtralProcessor
from transformers.testing_utils import require_vision
from transformers.testing_utils import require_read_token, require_vision
from transformers.utils import is_torch_available, is_vision_available
from ...test_processing_common import ProcessorTesterMixin
@@ -35,6 +35,7 @@ if is_vision_available():
@require_vision
@require_read_token
class Mistral3ProcessorTest(ProcessorTesterMixin, unittest.TestCase):
"""This tests Pixtral processor with the new `spatial_merge_size` argument in Mistral3."""