Skip to content

Commit 704fe46

Browse files
committed
fix issues
1 parent 0ec859a commit 704fe46

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

_unittests/ut_tasks/test_tasks_image_text_to_text.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,15 @@ def test_image_text_to_text_idefics(self):
3030
)
3131

3232
@hide_stdout()
33-
@requires_transformers("4.53")
33+
@requires_transformers("4.56")
3434
@requires_torch("2.7.99")
3535
def test_image_text_to_text_gemma3(self):
36+
"""
37+
If the model tails because of
38+
``if inputs_embeds[special_image_mask].numel() != image_features.numel():```,
39+
make sure this PR was merged:
40+
https://github.com/huggingface/transformers/pull/39962.
41+
"""
3642
# mid = "google/gemma-3-4b-it"
3743
mid = "tiny-random/gemma-3"
3844
data = get_untrained_model_with_inputs(mid, verbose=1, add_second_input=True)

0 commit comments

Comments
 (0)