riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-30 17:52:35 +06:00
Code Issues Actions 30 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats 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,