VLM: enable skipped tests (#35746)

* fix cached tests * fix some tests * fix pix2struct * fix
2025-02-12 12:55:46 +01:00
parent d6897b46bd
commit 8fc6ecba4f
10 changed files with 216 additions and 20 deletions
--- a/tests/models/aria/test_modeling_aria.py
+++ b/tests/models/aria/test_modeling_aria.py
@@ -82,14 +82,14 @@ class AriaVisionText2TextModelTester:
            moe_intermediate_size=4,
            moe_num_experts=4,
            moe_topk=2,
-            num_attention_heads=20,
+            num_attention_heads=8,
            num_experts_per_tok=3,
            num_hidden_layers=2,
-            num_key_value_heads=20,
+            num_key_value_heads=8,
            rope_theta=5000000,
            vocab_size=99,
            eos_token_id=2,
-            head_dim=2,
+            head_dim=4,
        ),
        is_training=True,
        vision_config=Idefics3VisionConfig(