From b2c80f8ab4099dfadb3cc75c270f6633f62b8325 Mon Sep 17 00:00:00 2001 From: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com> Date: Wed, 24 May 2023 17:31:25 +0100 Subject: [PATCH] [Whisper] Reduce batch size in tests (#23736) --- tests/models/whisper/test_modeling_whisper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/whisper/test_modeling_whisper.py b/tests/models/whisper/test_modeling_whisper.py index 98bbbb3214a7be..2be7f6884e72bb 100644 --- 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,