SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Sanchit Gandhi
2023-05-24 17:31:25 +01:00
committed by GitHub
parent 814de8fac7
commit d8222be57e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/whisper/test_modeling_whisper.py
View File

@@ -94,7 +94,7 @@ class WhisperModelTester:
def __init__( def __init__(
self, self,
parent, parent,
batch_size=13, batch_size=2,
seq_length=1500, seq_length=1500,
is_training=True, is_training=True,
use_labels=False, use_labels=False,
@@ -1537,7 +1537,7 @@ class WhisperEncoderModelTester:
def __init__( def __init__(
self, self,
parent, parent,
batch_size=13, batch_size=2,
seq_length=3000, seq_length=3000,
is_training=True, is_training=True,
use_labels=True, use_labels=True,