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_video/test_modeling_llava_next_video.py
+++ b/tests/models/llava_next_video/test_modeling_llava_next_video.py
@@ -91,7 +91,7 @@ class LlavaNextVideoVisionText2TextModelTester:
        },
        is_training=True,
        vision_config={
-            "image_size": 16,
+            "image_size": 8,
            "patch_size": 4,
            "num_channels": 3,
            "is_training": True,
@@ -125,10 +125,10 @@ class LlavaNextVideoVisionText2TextModelTester:
        self.batch_size = 3
        self.num_channels = 3
        self.image_size = 30
-        self.encoder_seq_length = 127
-        self.image_grid_pinpoints = [[32, 32]]
-        self.num_image_tokens = 88
-        self.num_video_tokens = 32
+
+        self.image_grid_pinpoints = [[16, 16]]
+        self.num_image_tokens = 24
+        self.num_video_tokens = 8
        self.seq_length = seq_length + self.num_image_tokens + self.num_video_tokens

    def get_config(self):