make test_eager_matches_sdpa_inference less flaky (#34512)

* try * try * try * try * try * try * update * update * update * update * update * update * update --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-10-31 18:34:00 +01:00
parent 294c170ff9
commit 114dd812dd
4 changed files with 45 additions and 54 deletions
--- a/tests/models/llava/test_modeling_llava.py
+++ b/tests/models/llava/test_modeling_llava.py
@@ -85,7 +85,7 @@ class LlavaVisionText2TextModelTester:
        },
        is_training=True,
        vision_config={
-            "image_size": 30,
+            "image_size": 8,
            "patch_size": 2,
            "num_channels": 3,
            "is_training": True,
@@ -118,9 +118,9 @@ class LlavaVisionText2TextModelTester:
        self.batch_size = 3
        self.num_channels = 3
        self.image_size = 336
-        self.encoder_seq_length = 232
-        self.num_image_tokens = 225
+        self.num_image_tokens = (self.vision_config["image_size"] // self.vision_config["patch_size"]) ** 2
        self.seq_length = seq_length + self.num_image_tokens
+        self.encoder_seq_length = self.seq_length

    def get_config(self):
        return LlavaConfig(