Skip to content

Commit d008cbf

Browse files
author
xiaoyunchong.xyc
committed
fix: Fun-ASR-Nano batch_size check - use len(data_in) instead of kwargs batch_size_ms
1 parent b9bcf9d commit d008cbf

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • examples/industrial_data_pretraining/fun_asr_nano

examples/industrial_data_pretraining/fun_asr_nano/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -470,7 +470,7 @@ def inference_prepare(
470470
):
471471
meta_data = {}
472472

473-
if kwargs.get("batch_size", 1) > 1:
473+
if isinstance(data_in, (list, tuple)) and len(data_in) > 1:
474474
raise NotImplementedError("batch decoding is not implemented")
475475

476476
contents = self.data_template(data_in[0])

0 commit comments

Comments
 (0)