Skip to content

GitLab

Unverified Commit d8222be5 authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

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

parent 814de8fa
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/whisper/test_modeling_whisper.py
View file @ d8222be5
...@@ -94,7 +94,7 @@ class WhisperModelTester: ...@@ -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: ...@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment