Skip to content

Commit c4be409

Browse files
committed
feat: change default batch size
1 parent 2eb27d6 commit c4be409

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

test_asr.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def main():
4646

4747

4848
def run_batched(
49-
instances: List[Dict[str, str | int]], batch_size: int = 64
49+
instances: List[Dict[str, str | int]], batch_size: int = 4
5050
) -> List[Dict[str, str | int]]:
5151
# split into batches
5252
results = []

test_nlp.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def main():
3636

3737

3838
def run_batched(
39-
instances: List[Dict[str, str | int]], batch_size: int = 64
39+
instances: List[Dict[str, str | int]], batch_size: int = 4
4040
) -> List[Dict[str, str | int]]:
4141
# split into batches
4242
results = []

test_vlm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def main():
6464

6565

6666
def run_batched(
67-
instances: List[Dict[str, str | int]], batch_size: int = 64
67+
instances: List[Dict[str, str | int]], batch_size: int = 44
6868
) -> List[Dict[str, str | int]]:
6969
# split into batches
7070
results = []

0 commit comments

Comments
 (0)