[Whisper] Reduce batch size in tests (#23736)

2023-05-24 17:31:25 +01:00
parent 814de8fac7
commit d8222be57e
1 changed files with 2 additions and 2 deletions
--- a/tests/models/whisper/test_modeling_whisper.py
+++ b/tests/models/whisper/test_modeling_whisper.py
@@ -94,7 +94,7 @@ class WhisperModelTester:
    def __init__(
        self,
        parent,
-        batch_size=13,
+        batch_size=2,
        seq_length=1500,
        is_training=True,
        use_labels=False,
@@ -1537,7 +1537,7 @@ class WhisperEncoderModelTester:
    def __init__(
        self,
        parent,
-        batch_size=13,
+        batch_size=2,
        seq_length=3000,
        is_training=True,
        use_labels=True,