Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions tests/models/smolvlm/test_processor_smolvlm.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@

from transformers import SmolVLMProcessor
from transformers.models.auto.processing_auto import AutoProcessor
from transformers.testing_utils import is_flaky, require_av, require_torch, require_vision
from transformers.testing_utils import require_av, require_torch, require_vision
from transformers.utils import is_vision_available

from ...test_processing_common import ProcessorTesterMixin
Expand Down Expand Up @@ -118,10 +118,14 @@ def get_split_image_expected_tokens(self, processor, image_rows, image_cols):
def tearDownClass(cls):
shutil.rmtree(cls.tmpdirname, ignore_errors=True)

@is_flaky # fails 15 out of 100, FIXME @raushan
@unittest.skip("flaky on main") # FIXME @raushan
def test_structured_kwargs_nested_from_dict_video(self):
super().test_structured_kwargs_nested_from_dict_video()

@unittest.skip("flaky on main") # FIXME @raushan
def test_structured_kwargs_nested_video(self):
super().test_structured_kwargs_nested_video()

def test_process_interleaved_images_prompts_no_image_splitting(self):
processor_components = self.prepare_components()
processor_components["tokenizer"] = self.get_component("tokenizer", padding_side="left")
Expand Down