Remove @slow for test_eager_matches_sdpa_inference (#34558)

* update * update * update * update * update * update * update * update * update * update * update --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-11-05 16:10:42 +01:00
parent 082e57e0d4
commit f2d5dfbab2
21 changed files with 271 additions and 626 deletions
--- a/tests/models/llava_next/test_modeling_llava_next.py
+++ b/tests/models/llava_next/test_modeling_llava_next.py
@@ -90,7 +90,7 @@ class LlavaNextVisionText2TextModelTester:
        },
        is_training=True,
        vision_config={
-            "image_size": 16,
+            "image_size": 8,
            "patch_size": 4,
            "num_channels": 3,
            "is_training": True,
@@ -123,10 +123,10 @@ class LlavaNextVisionText2TextModelTester:
        self.batch_size = 3
        self.num_channels = 3
        self.image_size = 30
-        self.encoder_seq_length = 95
-        self.image_grid_pinpoints = [[32, 32]]
-        self.num_image_tokens = 88
+        self.image_grid_pinpoints = [[16, 16]]
+        self.num_image_tokens = 24
        self.seq_length = seq_length + self.num_image_tokens
+        self.encoder_seq_length = self.seq_length

    def get_config(self):
        return LlavaNextConfig(